我使用H2O版本3.26.0.5训练了一个用于二元问题的GBM模型,用于预测正类的概率。我将模型文件保存为MOJO格式,并使用此文件在新数据上生成预测:
## 首先,重启R会话
### 加载模型
library(h2o)
h2o.init(nthreads = -1)
model <- h2o.import_mojo("path_to_mojo_file")
# 加载新的数据输入
input <- read_csv("path_to_new_data")
input_h2o <- as.h2o(input)
# 预测
predictions <- predict(model, input_h2o)
当我在我的电脑上运行这段代码时,得到的预测结果与在生产环境中使用相同的MOJO文件进行预测的结果不同。
这种情况应该发生在使用MOJO文件时吗?我以为一旦模型以MOJO格式保存后,就可以在任何环境中进行预测并得到相同的结果。有没有人知道为什么会发生这种情况?
回答:
最后我发现生产环境的脚本中有一个错误。在修复后,预测结果变得非常接近。