【发布时间】:2020-03-18 21:06:46
【问题描述】:
我使用 H2O 版本 3.26.0.5 在二元问题中训练 GBM 模型,以预测正类的概率。我将模型文件保存为 MOJO 并使用此文件在新数据中生成预测:
## first, restart R session ##
# load the model
library(h2o)
h2o.init(nthreads = -1)
model <- h2o.import_mojo("path_to_mojo_file")
# load the new data input
input <- read_csv("path_to_new_data")
input_h2o <- as.h2o(input)
# predictions
predictions <- predict(model, input_h2o)
当我在我的计算机上运行它时,我得到的预测与我在生产环境中使用相同的 MOJO 文件进行预测时得到的预测不同。
MOJO 文件会发生这种情况吗?我相信一旦模型以 MOJO 格式保存,您可以在任何环境下进行预测并获得相同的结果。有谁知道为什么会这样?
【问题讨论】:
-
您好,这是意料之外的。你能提供一个可重现的例子吗?
标签: r machine-learning prediction h2o gbm