diff --git a/code/bolsonaro/data/dataset_loader.py b/code/bolsonaro/data/dataset_loader.py index a80d3ca5132427202a56a63ce94659b5c3d713e3..fc0caa8922c36bc5b8e420d6cde7317119e8cd67 100644 --- a/code/bolsonaro/data/dataset_loader.py +++ b/code/bolsonaro/data/dataset_loader.py @@ -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))