为啥当 cv=5 时 cross_val_predict 只返回单个预测数组

Posted 2023-03-12

【中文标题】为啥当 cv=5 时 cross_val_predict 只返回单个预测数组

【英文标题】：Why cross_val_predict returns only single prediction array when cv=5为什么当 cv=5 时 cross_val_predict 只返回单个预测数组

【发布时间】：2018-11-26 15:11:35

【问题描述】：

当我跑步时

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

我从中得到一个一维数组：

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

每个折叠不应该有一个预测数组吗？

类似于

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

返回 5 个项目数组。每折一分。

【参考方案1】：

cross_val_predict 对您感兴趣的完整数据集进行 ouf-of-fold 预测。您的数据集分为 k 折叠，模型在 k-1 折叠上进行训练，并预测保留保持为输出生成和存储。详情请见this answer on SO。