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

Question

从R包级别使用h2o时，我可以将用h2o训练的模型导出到 Java 代码吗？

Answer 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）。