Merge pull request #89 from nyanp/fix/scikit-learn-1.0
Support scikit-learn 1.0feature/enhance-averaging
commit
4758343cac
|
@ -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):
|
||||
"""
|
||||
|
|
|
@ -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',
|
||||
|
|
Loading…
Reference in New Issue