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方法对训练集或测试集进行切分。