Description Usage Arguments Value Examples
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.
1 | map_h2o_encoding(h2o_trees_R)
|
h2o_trees_R |
|
list
containing split conditions to reach terminal nodes for
each tree.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 | 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)
|
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.