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

Adding a new regression dataset

parent 36e191bc
No related branches found
No related tags found
1 merge request!15Resolve "Adding new datasets"
......@@ -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))
......
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