同时调用SKLearn的cross_val_score和cross_val_predict?

时间:2016-08-16 14:30:12

标签: python performance machine-learning scikit-learn

我正在通过cross_val_score对象传递GridSearchCV来运行嵌套交叉验证。然后我跟进cross_val_predict以获得用于绘图的模型预测。像这样:

gs = GridSearchCV(mymodel, myparams)
score = cross_val_score(gridsearch, X_train, y_train)
prediction = cross_val_predict(gs, X_train, y_train)

这似乎在计算上是多余的;有没有办法从cross_val_score获得交叉验证的预测,或者我是否需要手动迭代CV对象的折叠才能一步完成?

1 个答案:

答案 0 :(得分:0)

根据我在文档中看到的内容,看来cross_val_predictreshape是通过不同的过程获得其值的,因此不建议将两者结合使用。

https://scikit-learn.org/stable/modules/cross_validation.html#obtaining-predictions-by-cross-validation