Как я могу получить одно дерево из случайного леса H2O?

Question

Я делаю случайный лес со следующим кодом:

rf_md <- h2o.randomForest(training_frame = train_h,
                          nfolds = 5,
                          y = y,
                          ntrees = 500,
                          stopping_rounds = 10,
                          stopping_metric = "RMSE",
                          score_each_iteration = TRUE,
                          stopping_tolerance = 0.0001,
                          seed = 1234)

На выходе получается объект с типом: h2oregressionmodel. Я хочу получить одно дерево из леса, поэтому я обнаружил, что функция getTree() очень хорошо работает с randomForest объектами. Как я могу преобразовать свой фактический вывод в randomForest объект? или Могу ли я получить тот же случайный лес с другой функцией, вывод которой - randomForest объект?

Ценим вашу помощь

Answer 1

H2O имеет аналогичную функцию h2o.getModelTree, которую можно использовать как для моделей GBM, так и для моделей с произвольным лесом (см. документы );в вашем случае для выбора, скажем, дерева № 3 оно должно быть:

tree <- h2o.getModelTree(model=rf_md, tree_number=3)