返回基于交叉验证的预测的回调函数

此回调函数会保存所有测试折叠的预测结果，并允许保存折叠对应的模型。

用法

xgb.cb.cv.predict(save_models = FALSE, outputmargin = FALSE)

参数

save_models

一个标志，指示是否保存折叠对应的模型。

outputmargin

是否保存边际预测（效果与将此参数传递给 predict.xgb.Booster 相同）。

返回值

一个 xgb.Callback 对象，可以传递给 xgb.cv()，但不能传递给 xgb.train()。

详情

预测结果保存在 pred 元素中，它是一个向量或矩阵，取决于每行数据对应的预测输出数量。预测结果的顺序与原始数据集中的行顺序一致。请注意，当在 xgb.cv() 中提供自定义 folds 列表时，仅当此列表是 k 组不重叠的索引集合（如标准 k 折交叉验证）时，预测结果才能正确返回。当用户提供的折叠包含重叠索引时（例如随机抽样分割），预测结果将没有意义。当训练数据集中的某些索引未包含在用户提供的 folds 中时，其预测值将为 NA。