首页 > 解决方案 > XGBoost 提取训练预测

问题描述

在非常嘈杂的数据集上生成模型时,我需要提取最终 XGBoost 模型在训练集上所做的预测。我知道我可以从 xgb.cv 中提取不折叠的预测,但是我不能用它来预测保留的测试集。

我明白这不是一个常见的用例,但有必要找出哪些样本代表训练集中的异常值,以便进一步探索和以后的模型堆叠。

有没有办法显示最终模型对训练集所做的预测?我会想,因为我可以在每一轮都出错,但我找不到方法。

标签: rxgboost

解决方案


如果我误解了这个问题,请原谅我,但predict()不起作用?

model = xgboost(data = data$train, nround = 20,param=param)
training_predictions = predict(model, data$train)

另外,我在无法安装 XGB 的机器上,但我很确定模型对象将最终预测作为元素之一。


推荐阅读