R/models.R
h2o.getModelTree.Rd
Usage example: airlines.data <- h2o.importFile(path = '/path/to/airlines_train.csv') gbm.model = h2o.gbm(x=c("Origin", "Dest", "Distance"),y="IsDepDelayed",training_frame=airlines.data ,model_id="gbm_trees_model") tree <-h2o.getModelTree(gbm.model, 1, 1);
h2o.getModelTree(model, tree_number, tree_class = NA)
model | Model with trees |
---|---|
tree_number | Number of the tree in the model to fetch, starting with 1 |
tree_class | Name of the class of the tree (if applicable). This value is ignored for regression and binomial response column, as there is only one tree built. As there is exactly one class per categorical level, name of tree's class equals to the corresponding categorical level of response column. |
Returns an H2OTree object with detailed information about a tree.