R XGBoost - xgb.save 或 xgb.load 数据丢失

Question

在 R 中训练 XGBoost 模型后，我会看到一个名为 xgb 的模型对象，它是一个 7 的列表。

当我使用 xgb.save 保存模型，然后使用 xgb.load 重新加载时，我会看到一个似乎“较小”的模型对象，它是2 的列表。

显然我无法共享代码，因为您需要大量的训练数据，所以我真正能展示的只是变量编辑器的图片。

下面是模型对象xgb，它是训练后的原始模型，与模型对象test1(它是相同的模型，但保存并重新加载):

为什么会发生这种情况？我在保存/加载模型时是否会丢失有值(value)的信息？

感谢任何帮助。

Answer 1

也许晚了，但我遇到了同样的问题并找到了解决方案。

将 xgb-model 保存为“rds”不会丢失任何信息，并且当我测试它时，重新加载的模型 xgb_ 确实会生成与原始 xgb 相同的预测值。希望有帮助!

saveRDS(xgb, "model.rds")
xgb_ <- readRDS("model.rds")
all.equal(xgb, xgb_)