Skip to content
Snippets Groups Projects

Resolve "Adding new datasets"

Merged Leo Bouscarrat requested to merge 17-adding-new-datasets into master
1 file
+ 3
0
Compare changes
  • Side-by-side
  • Inline
@@ -110,6 +110,9 @@ class DatasetLoader(object):
elif name == 'kr-vs-kp':
dataset_loading_func = change_binary_func_openml('kr-vs-kp')
task = Task.BINARYCLASSIFICATION
elif name == 'kin8nm':
X, y = fetch_openml('kin8nm', return_X_y=True, random_state=dataset_parameters.random_state)
task = Task.REGRESSION
else:
raise ValueError("Unsupported dataset '{}'".format(name))
Loading