avoid using deprecated load_boston() for test

2021-10-30 12:42:34 +09:00 · 2021-10-30 12:42:34 +09:00 · 3ad1cc8ce2
parent 0a48a76385
commit 3ad1cc8ce2
1 changed files with 1 additions and 3 deletions
--- a/tests/experiment/test_hyperparameter_tuner.py
+++ b/tests/experiment/test_hyperparameter_tuner.py
@ -16,9 +16,7 @@ def _check_parameter_tunes(params, x, y):


 def test_regression_problem_parameter_tunes():
-    dataset = datasets.load_boston()
-    x = pd.DataFrame(dataset.data, columns=dataset.feature_names)
-    y = pd.Series(dataset.target)
+    x, y = datasets.load_diabetes(return_X_y=True, as_frame=True)
    params = {
        'objective': 'regression',
        'metric': 'rmse',