R H2o-GBM模型中的树分裂

R H2o-GBM模型中的树分裂,r,h2o,gbm,R,H2o,Gbm,通过运行命令 m <- h2o.getModel("depth_grid_model_4") h2o.varimp(m) m您可以使用h2o将模型导出为POJO。下载\u POJO(),然后查看文件中每个树的完整细节。有一个工具可以为h2o-3 MOJO模型创建可视化效果。请参阅此处的完整文档: 使用R创建并下载MOJO: library(h2o) h2o.init() df <- h2o.importFile("http://s3.amazonaws.com/h2o-pu

通过运行命令

m <- h2o.getModel("depth_grid_model_4")
h2o.varimp(m)

m您可以使用
h2o将模型导出为POJO。下载\u POJO()
,然后查看文件中每个树的完整细节。

有一个工具可以为h2o-3 MOJO模型创建可视化效果。请参阅此处的完整文档:


使用R创建并下载MOJO:

library(h2o)
h2o.init()
df <- h2o.importFile("http://s3.amazonaws.com/h2o-public-test-data/smalldata/airlines/allyears2k_headers.zip")
model <- h2o.gbm(model_id = "model",
                training_frame = df,
                x = c("Year", "Month", "DayofMonth", "DayOfWeek", "UniqueCarrier"),
                y = "IsDepDelayed",
                max_depth = 3,
                ntrees = 5)
h2o.download_mojo(model, getwd(), FALSE)

3.22.0.1中在H2O中添加了新的Tree API。它允许您从H2O中任何基于树的模型将树提取到R/Python对象中(有关详细信息,请参阅):


tree如果您的问题得到解决,请接受下面的答案。谢谢关于命令行,我在这里遇到了一些简单的问题?我需要从哪个目录运行此操作?我当前的工作目录?还是特定的h2o文件夹?我收到以下消息:错误:无法找到或加载主类hex.genmodel.tools.PrintMojo“java-cp/path/to/your/h2o.jar”。你需要知道你的h2o.jar在哪里。它将位于安装h2o包的R或Python目录中,或者您可以从h2oai.download->h2o latest stable release->download h2o下载它。
# (For MacOS: brew install graphviz)
java -cp h2o.jar hex.genmodel.tools.PrintMojo --tree 0 -i model.zip -o model.gv
dot -Tpng model.gv -o model.png
open model.png
tree <- h2o.getModelTree(model = airlines.model, tree_number = 1, tree_class = "NO")