arbre se divise en un H2o GBM Modèle

Question

En exécutant les commandes,

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

Je suis en mesure d'afficher les performances du modèle, ainsi que l'importance variable.

Comment afficher les divisions utilisées dans chaque arbre du modèle GBM?

Merci

Answer 1

Vous pouvez exporter le modèle comme POJO avec h2o.download_pojo() puis regarder tous les détails de chaque arbre dans le fichier.

Answer 2

Il existe un outil pour créer des visualisations pour les modèles H2O-3 MOJO. Voir la documentation complète ici:

• http://docs.h2o.ai/h2o/latest-stable/h2o-genmodel/javadoc/overview-summary.html#viewing-a-mojo

---

Utilisez R pour créer et télécharger un 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) 

Exécutez l'outil PrintMojo (emballé à l'intérieur h2o.jar) sur la ligne de commande pour créer un fichier .png. Vous devez télécharger la dernière version stable H2O-3 de http://www.h2o.ai/download/ et exécuter l'outil PrintMojo à partir de la ligne de commande.

# (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