Визуализируйте деревья в модели H2O XGBoost

Question

Я смотрел на этот ответ, чтобы визуализировать модель дерева повышения градиента в H2O, там говорится, что метод на GBM может быть применен и к XGBoost:
Поиск вклада каждой функции в создание конкретного прогноза с помощью h2oмодель ансамбля

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

Но когда я пытаюсь использовать метод, описанный в H2O XGBoost MOJO, он не работает.

Я проверяю исходный код hex.genmodel.tools.PrintMojo: https://github.com/h2oai/h2o-3/blob/master/h2o-genmodel/src/main/java/hex/genmodel/tools/PrintMojo.java

кажется, что он может работать только на моделях randomforest и GBM, но не на модели XGBoost.

Есть ли кто-нибудь, кто знает, как визуализировать деревья в модели H2O XGBoost?Спасибо!

Answer 1

Я наконец-то нашел решение, которое, похоже, не документировано для XGBoost, но оно действительно такое же, как и для других алгоритмов, связанных с деревьями.

Просто запустите эту команду, чтобы сгенерировать первые 50 деревьев из вашей модели:

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

Answer 2

Это функция, которую H2O в настоящее время добавляет, вы можете отслеживать ее прогресс здесь: https://0xdata.atlassian.net/browse/PUBDEV-5743.

Обратите внимание, что в заявке есть предложение о том, как визуализировать деревья с помощью собственного xgboost.