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