map_h2o_encoding: Get temrinal node paths in terms of L/R directions and...

In richardangell/h2oTreeHelpR: h2o Tree Helper

Description

This function takes the output from H2OTreeConvertR::h2o_tree_convertR and for each tree returns the terminal node decision paths in terms of L/R directions and variable conditions. This can be used to map the output from h2o::h2o.predict_leaf_node_assignment to variable condtions.

Usage

map_h2o_encoding(h2o_trees_R)

Arguments

h2o_trees_R

list output from the H2OTreeConvertR::h2o_tree_convertR function.

Value

list containing split conditions to reach terminal nodes for each tree.

Examples

library(h2o)

h2o.init()

prostate.hex = h2o.uploadFile(path = system.file("extdata",
                                                 "prostate.csv",
                                                 package = "h2o"),
                              destination_frame = "prostate.hex")

prostate.hex$RACE <- as.factor(prostate.hex$RACE)

prostate.hex$DPROS <- as.factor(prostate.hex$DPROS)

prostate.gbm = h2o.gbm(x = c("AGE", "RACE", "DPROS", "DCAPS", "PSA", "VOL", "GLEASON"),
                       y = "CAPSULE",
                       training_frame = prostate.hex,
                       ntrees = 5,
                       max_depth = 5,
                       learn_rate = 0.1)

h2o_trees <- H2OTreeConvertR::h2o_tree_convertR(prostate.gbm)

h2o_terminal_node_encoding_mapping <- map_h2o_encoding(h2o_trees)

richardangell/h2oTreeHelpR documentation built on July 3, 2019, 7:35 a.m.