diff --git a/code/bolsonaro/data/dataset_loader.py b/code/bolsonaro/data/dataset_loader.py index fc0caa8922c36bc5b8e420d6cde7317119e8cd67..df132fc46a012184ab02adeaea1c536e3ad6be4d 100644 --- a/code/bolsonaro/data/dataset_loader.py +++ b/code/bolsonaro/data/dataset_loader.py @@ -31,13 +31,14 @@ class DatasetLoader(object): dataset_names = ['boston', 'iris', 'diabetes', 'digits', 'linnerud', 'wine', 'breast_cancer', 'olivetti_faces', '20newsgroups_vectorized', 'lfw_people', - 'lfw_pairs', 'covtype', 'rcv1', 'california_housing', 'diamonds'] + 'lfw_pairs', 'covtype', 'rcv1', 'california_housing', 'diamonds', 'steel-plates', + 'kr-vs-kp', 'kin8nm'] dataset_seed_numbers = {'boston':15, 'iris':15, 'diabetes':15, 'digits':5, 'linnerud':15, 'wine':15, 'breast_cancer':15, 'olivetti_faces':15, '20newsgroups_vectorized':3, 'lfw_people':3, 'lfw_pairs':3, 'covtype':3, 'rcv1':3, 'california_housing':3, - 'diamonds': 15} + 'diamonds': 15, 'steel-plates':15, 'kr-vs-kp':15, 'kin8nm':15} @staticmethod def load(dataset_parameters):