sklearn中GridSearchCV输出及KFold返回值的技术问询
关于sklearn嵌套交叉验证的两个疑问解答
问题1:传入cross_val_score的GridSearchCV实例clf,是待训练的多超参数分类器集合,还是已训练好的最优参数分类器?
- 这个
clf是待训练的多超参数分类器集合。把它传入cross_val_score后,外层交叉验证的每一轮都会基于当前数据子集,先让GridSearchCV在子集的训练段完成超参数搜索——遍历所有指定的参数组合,通过内部交叉验证选出最优参数,再用带最优参数的分类器在子集的测试段做评估。整个流程里,clf不会提前完成训练,而是在交叉验证的每一轮中完成训练和参数筛选。
问题2:KFold实例返回的是用于后续数据划分的k折索引吗?
- 没错。
KFold是一个数据划分生成器,调用它的split()方法并传入数据集后,会得到一个迭代器,每次迭代都会输出一组(训练集索引,测试集索引),这些索引就是用来划分原始数据集的,你可以直接用它们从原数据中提取对应的训练、测试样本。
内容的提问来源于stack exchange,提问作者user21840125
相关产品推荐
相关产品推荐

