From 96f51d9cfbda7214fb6005f83bfbd938a42c2b7d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?L=C3=A9o=20Bouscarrat?= <leo.bouscarrat@euranova.eu>
Date: Wed, 4 Mar 2020 18:06:18 +0100
Subject: [PATCH] Adding the last datasets added to dataset_names and
dataset_seed_numbers
---
code/bolsonaro/data/dataset_loader.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/code/bolsonaro/data/dataset_loader.py b/code/bolsonaro/data/dataset_loader.py
index fc0caa8..df132fc 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):
--
GitLab