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"
This commit is part of merge request !15. Comments created here will be created in the context of that merge request.
Hide whitespace changes
Inline Side-by-side
code/bolsonaro/data/dataset_loader.py
+ 3
− 2
Edit View file @ 96f51d9c
...@@ -31,13 +31,14 @@ class DatasetLoader(object): ...@@ -31,13 +31,14 @@ class DatasetLoader(object):
dataset_names = ['boston', 'iris', 'diabetes', 'digits', 'linnerud', 'wine', dataset_names = ['boston', 'iris', 'diabetes', 'digits', 'linnerud', 'wine',
'breast_cancer', 'olivetti_faces', '20newsgroups_vectorized', 'lfw_people', '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, dataset_seed_numbers = {'boston':15, 'iris':15, 'diabetes':15, 'digits':5,
'linnerud':15, 'wine':15, 'breast_cancer':15, 'olivetti_faces':15, 'linnerud':15, 'wine':15, 'breast_cancer':15, 'olivetti_faces':15,
'20newsgroups_vectorized':3, 'lfw_people':3, '20newsgroups_vectorized':3, 'lfw_people':3,
'lfw_pairs':3, 'covtype':3, 'rcv1':3, 'california_housing':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 @staticmethod
def load(dataset_parameters): def load(dataset_parameters):
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment