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

我制作了一个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

值

0.85

。但我的问题是：如何将预测的varToBePredicted值添加到

data

（这是一个

data.table

）中？我想这样做，以便我可以绘制观察值与预测值。

假设您的结果位于名为

results

的对象中，请共享

dput（head（results））

和

dput（head（data））

（您的data.table）。投票结束时“太宽”。不幸的是，没有玩具数据很难帮助。。。

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

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