fix constructor of StratifiedGroupKFold
parent
1fa9a3e9cf
commit
0a48a76385
|
@ -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):
|
||||
"""
|
||||
|
|
Loading…
Reference in New Issue