From 3bafa452ce9382e18a0e87d24c3f7f460d146062 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:04:58 +0100
Subject: [PATCH] Adding a new regression dataset
---
code/bolsonaro/data/dataset_loader.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/code/bolsonaro/data/dataset_loader.py b/code/bolsonaro/data/dataset_loader.py
index a80d3ca..fc0caa8 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))
--
GitLab