Merge pull request #89 from nyanp/fix/scikit-learn-1.0

Support scikit-learn 1.0

Author: nyanp

nyaggle/validation/split.py

@@ -46,7 +46,7 @@ def check_cv(cv: Union[int, Iterable, BaseCrossValidator] = 5,
         else:
             return KFold(cv, shuffle=True, random_state=random_state)
 
-    return model_selection.check_cv(cv, y, stratified)
+    return model_selection.check_cv(cv, y, classifier=stratified)
 
 
 class Take(BaseCrossValidator):
@@ -380,8 +380,7 @@ class StratifiedGroupKFold(_BaseKFold):
 
     def __init__(self, n_splits: int = 3, shuffle: bool = False,
                  random_state: Optional[Union[int, np.random.RandomState]] = None):
-        super(StratifiedGroupKFold, self).__init__(n_splits, shuffle,
-                                                   random_state)
+        super().__init__(n_splits, shuffle=shuffle, random_state=random_state)
 
     def _make_test_folds(self, X, y=None, groups=None):
         """

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',