我可以將使用 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）。