Skip to content
Snippets Groups Projects
Commit 96f51d9c authored by Léo Bouscarrat's avatar Léo Bouscarrat
Browse files

Adding the last datasets added to dataset_names and dataset_seed_numbers

parent 3bafa452
No related branches found
No related tags found
1 merge request!15Resolve "Adding new datasets"
......@@ -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):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment