Skip to content
Snippets Groups Projects

Resolve "integration-sota"

Merged Charly Lamothe requested to merge 15-integration-sota into master
21 files
+ 1805
345
Compare changes
  • Side-by-side
  • Inline
Files
21
from bolsonaro.data.dataset import Dataset
from bolsonaro.data.dataset import Dataset
from bolsonaro.data.dataset_parameters import DatasetParameters
from bolsonaro.data.dataset_parameters import DatasetParameters
from bolsonaro.data.task import Task
from bolsonaro.data.task import Task
from bolsonaro.utils import change_binary_func_load, change_binary_func_openml
from bolsonaro.utils import change_binary_func_load, change_binary_func_openml, binarize_class_data
from sklearn.datasets import load_boston, load_iris, load_diabetes, \
from sklearn.datasets import load_boston, load_iris, load_diabetes, \
load_digits, load_linnerud, load_wine, load_breast_cancer
load_digits, load_linnerud, load_wine, load_breast_cancer
@@ -81,7 +81,9 @@ class DatasetLoader(object):
@@ -81,7 +81,9 @@ class DatasetLoader(object):
elif name == 'lfw_pairs':
elif name == 'lfw_pairs':
dataset = fetch_lfw_pairs()
dataset = fetch_lfw_pairs()
X, y = dataset.data, dataset.target
X, y = dataset.data, dataset.target
task = Task.MULTICLASSIFICATION
possible_classes = sorted(set(y))
 
y = binarize_class_data(y, possible_classes[-1])
 
task = Task.BINARYCLASSIFICATION
elif name == 'covtype':
elif name == 'covtype':
X, y = fetch_covtype(random_state=dataset_parameters.random_state, shuffle=True, return_X_y=True)
X, y = fetch_covtype(random_state=dataset_parameters.random_state, shuffle=True, return_X_y=True)
task = Task.MULTICLASSIFICATION
task = Task.MULTICLASSIFICATION
Loading