R:根据h2o.randomForest()和h2o.gbm()绘制树木
正在寻找一种有效的方法,在rstudio、H2O的Flow或H2O的RF和GBM模型的本地html页面中绘制树,类似于下面链接中图像中的树具体来说,您如何为下面的代码生成的对象(拟合模型)rf1和gbm2绘制树,可能是通过解析h2o。下载\u pojo(rf1)或h2o。下载\u pojo(gbm1)?R:根据h2o.randomForest()和h2o.gbm()绘制树木,r,data-visualization,random-forest,h2o,gbm,R,Data Visualization,Random Forest,H2o,Gbm,正在寻找一种有效的方法,在rstudio、H2O的Flow或H2O的RF和GBM模型的本地html页面中绘制树,类似于下面链接中图像中的树具体来说,您如何为下面的代码生成的对象(拟合模型)rf1和gbm2绘制树,可能是通过解析h2o。下载\u pojo(rf1)或h2o。下载\u pojo(gbm1)? ###以下两个命令删除以前为R安装的所有H2O软件包。 #如果(“包:h2o”%in%search()){detach(“包:h2o”,unload=TRUE)} #如果%rownames(i
###以下两个命令删除以前为R安装的所有H2O软件包。
#如果(“包:h2o”%in%search()){detach(“包:h2o”,unload=TRUE)}
#如果%rownames(installed.packages())中的(“h2o”%{remove.packages(“h2o”)}
##接下来,我们下载H2O依赖的软件包。
#pkgs我认为这可能是您正在寻找的解决方案
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)
现在从下载最新的稳定h2o版本,并从命令行运行PrintMojo工具
java -cp h2o.jar hex.genmodel.tools.PrintMojo --tree 0 -i model.zip -o model.gv
dot -Tpng model.gv -o model.png
打开model.png
更多信息:3.22.0.1(2018年10月)中引入的新树API改变了整个H2O树可视化游戏。一般工作流可能如下所示:
详细的代码示例可以在这里找到:。提供了可复制的示例。感谢您添加可复制的示例。我们可以将此迁移到现在。如果你等一等,它应该很快就到了。有没有一种方法可以像用图形的方式可视化最后一棵树那样来绘制呢????这是什么意思?我对这里的一个解决方案也很感兴趣。源代码:你如何用颜色/形状来表示末端节点的分类?
java -cp h2o.jar hex.genmodel.tools.PrintMojo --tree 0 -i model.zip -o model.gv
dot -Tpng model.gv -o model.png