def load_dataset(file_name): dataset = Dataset(logger, file_name) columns = ['date', 'security', 'feature'] dataset.drop_na(columns) dataset.factorize_label() dataset.sort_date() return dataset