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

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

用法

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。