Visualize trees in H2O XGBoost model

Question

I was looking at this answer to visualize the gradient boosting tree model in H2O, it says the method on GBM can be applied to XGBoost as well:
Finding contribution by each feature into making particular prediction by h2o ensemble model

http://docs.h2o.ai/h2o/latest-stable/h2o-docs/productionizing.html

But when I try to use the method it mentioned on H2O XGBoost MOJO, it fails.

I check the source code of hex.genmodel.tools.PrintMojo:https://github.com/h2oai/h2o-3/blob/master/h2o-genmodel/src/main/java/hex/genmodel/tools/PrintMojo.java

it seems like it can only work on randomforest and GBM models, but not XGBoost model.

Is there anyone who knows how to visualize trees in H2O XGBoost model? Thanks!

score 1 · Answer 1 · answered Sep 27 '18 at 18:21

1

This is a feature H2O is currently adding, you can track its progress here: https://0xdata.atlassian.net/browse/PUBDEV-5743.

Note that in the ticket there is a suggestion in the comments on how to visualize the trees with native xgboost.

answered Sep 27 '18 at 18:21

Lauren

5,640
1
13
19

1

Thanks! Do you know how to output the H2O XGBoost model to native XGBoost model? – Zhirui Wang Sep 28 '18 at 01:45

score 0 · Answer 2 · answered Apr 19 '19 at 12:29

I finally found the solution, that seem not documented for XGBoost, but it is indeed the same as for other trees-related algorithms.

Just run this command to generate the first 50 trees from your model:

for tn in {1..50}
do
   java -cp h2o-3.24.0.1/h2o.jar hex.genmodel.tools.PrintMojo --tree $tn -i <your mojo model> -o XGBOOST_$tn.gv
   dot -Tpng  XGBOOST_$tn.gv -o xgboost_$tn.png
done

Visualize trees in H2O XGBoost model

2 Answers2