Как визуализировать дерево H2O? - PullRequest
Новая
       57

Как визуализировать дерево H2O?

0 голосов
/ 08 июля 2019

У меня есть df data_categorical и модель model.
Я преобразовал свою df в h2o frame с
data = h2o.H2OFrame(data_categorical)

и обучил свою модель с 

model = H2ORandomForestEstimator(ntrees=1, max_depth=20, nfolds=10)
# Train model
model.train(x=training_columns, y=response_column, training_frame=train)

Я пытаюсь визуализировать созданное дерево (обратите внимание, что мне нужно только одно дерево), но я не могу этого сделать.
Я скачал файл mojo с
model.download_mojo(path,get_genmodel_jar=True)
Но я не знаю, что делать дальше

1 Ответ

1 голос
/ 08 июля 2019

Вы можете использовать рабочий процесс, описанный в документации ниже: 

java -cp h2o.jar hex.genmodel.tools.PrintMojo --tree 0 -i model.zip -o model.gv -f 20 -d 3
dot -Tpng model.gv -o model.png
open model.png
...