fix constructor of StratifiedGroupKFold

pull/89/head
Taiga Noumi 2021-10-30 12:31:17 +09:00
parent 1fa9a3e9cf
commit 0a48a76385
1 changed files with 1 additions and 2 deletions

View File

@ -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):
"""