我可以将使用h2o训练的模型导出到Java代码吗?

时间:2017-02-13 10:07:05

标签: java r production-environment h2o

h2o包的级别使用h2o时,是否可以将使用R培训的模型导出为Java代码?

1 个答案:

答案 0 :(得分:1)

是。这正是 POJO 的用途。在您的R脚本中,如果m是您的模型,那么您可以使用以下内容获取POJO:

h2o.download_pojo(m, "/path/to/save/in/");

这将选择文件名,并在您提供的目录中创建一个java文件。默认情况下,它还会下载您需要的jar文件。

如果不希望保存到本地文件系统,请不要给出路径(即只做h2o.download_pojo(m)),它会将java类输出到您的R会话,在那里您可以捕获输出,并做一些事情。

另请查看h2o.download_mojo(),它提供了不同的格式。它适用于树模型,它可以变得非常大(实际上它目前只支持随机森林,GBM和GLM)。

有关POJO和MOJO的更多信息:http://docs.h2o.ai/h2o/latest-stable/h2o-docs/pojo-quick-start.html