sklearn.model_selection.check_cv?

sklearn.model_selection.check_cv(cv=5, y=None, *, classifier=False)

[源碼]

用于構建交叉驗證器的輸入檢查器實用程序。

參數 說明
cv int, cross-validation generator or an iterable, default=None
確定交叉驗證切分策略。可能輸入的cv是：
-None，使用默認的5折交叉驗證，
-integer，指定折疊數。
- CV splitter，
-可迭代得到訓練集和測試集的切分作為數組的索引。

對于integer或None，如果分類器為True且y為二分類或多類，使用StratifiedKFold。在所有其他情況下，使用KFold。

有關可在此處使用的各種交叉驗證策略，請參閱用戶指南。

在0.22版中更改：cv默認值從3折更改為5折。
y array-like, default=None
監督學習問題的目標變量。
classifier bool, default=False
任務是否為分類任務，在這種情況下將使用分層的KFold。
返回值 說明
check_cv a cross-validator instance.
返回值是一個交叉驗證器，可通過split方法對訓練集或測試集進行切分。