fix argument of check_cv

2021-10-30 12:27:01 +09:00 · 2021-10-30 12:27:01 +09:00 · 1fa9a3e9cf
parent 32bffe86bb
commit 1fa9a3e9cf
1 changed files with 1 additions and 1 deletions
--- a/nyaggle/validation/split.py
+++ b/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):