如何从 mojo zip 文件中检索 H2O 模型？答案

【问题标题】：How to retrieve H2O model from mojo zip file?如何从 mojo zip 文件中检索 H2O 模型？
【发布时间】：2019-12-16 15:33:47
【问题描述】：

我使用 model.download_mojo(path="path", get_genmodel_jar=True) 保存了 H2O 模型。我想再次检索该模型以在 jupyter notebook 中使用。我怎样才能做到这一点？谢谢。

【问题讨论】：

标签： python h2o mojo

【解决方案1】：

你可以这样做：

data = h2o.import_file(path='training_dataset.csv')
original_model = H2OGeneralizedLinearEstimator()
original_model.train(x = ["Some column", "Another column"], y = "response", training_frame=data)

path = '/path/to/model/directory/model.zip'
original_model.download_mojo(path)

然后在新笔记本中执行以下操作：

path = '/path/to/model/directory/model.zip'
imported_model = h2o.import_mojo(path)
new_observations = h2o.import_file(path='new_observations.csv')
predictions = imported_model.predict(new_observations)

[取自文档中的此页面：

http://docs.h2o.ai/h2o/latest-stable/h2o-docs/data-science/mojo_import.html]

【讨论】：

谢谢 TomKraljevic，我可以导入模型，但这是一个随机森林模型。后来我将此模型传递给 H2OTree 方法。但我得到一个错误，这不是基于树的模型。我认为它是作为 MOJO 模型导入的。有什么方法可以导入基于树的模型。
您可以调用 saveModel 和 loadModel，而不是导出 MOJO。但 H2O-3 软件版本号需要相同。
是的，但是这个模型过去是作为 mojo 导出的。