当cv = 5时,为什么cross_val_predict仅返回单个预测数组

时间:2018-06-17 15:45:58

标签: scikit-learn

当我跑步时

cross_val_predict(a_clf, X_train, y_train, cv=5)

我从中得到一个1d数组:

array([False, False,  True, ..., False, False, False])

每个折叠不应该有一系列的预测吗?

相似
cross_val_score(a_clf, X_train, y_train, cv=5)

返回5项数组。每次折叠一次。

1 个答案:

答案 0 :(得分:2)

return对您感兴趣的完整数据集进行了多次预测。您的数据集被分为update()个折叠,模型以k-1折叠进行训练,并预测产生保持保持并存储输出。有关详细信息,请参阅this answer on SO