如何在现有数据框中保存H2O.GBM的预测值？

Question

我制作了一个H2O模型来预测varToBePredicted的值：

data <- h2o.importFile(file)
split <- h2o.splitFrame(data, ratios = c(.70, .15))

gbm <- h2o.gbm(
  training_frame = split[[1]],
  validation_frame = split[[2]],
  x = c(setdiff(names(data), allExceptThis)),
  y = 'varToBePredicted',
  ntrees = 1000,
  max_depth = 2)

model_path <- h2o.saveModel(object = gbm, path = getwd(), force=TRUE)

当我通过[]打印此程序的R2时>

print(h2o.r2(h2o.performance(gbm, newdata=split[[3]])))
我得到R2的值为0.85。但是我的问题是：如何将varToBePredicted的预测值添加到data（这是data.table）？我想要这样做，以便可以绘制观察值与预测值的关系。

我制作了一个H2O模型来预测varToBePredicted的值：数据

Answer 1

as.data.frame(h2o.predict(gbm, newdata=split[[3]]))