Description Usage Arguments Value Examples
This function uses recursion to append all split conditions that lead to terminal nodes.
1 | get_terminal_node_paths(df, node, current_str = NULL, unique_expr = TRUE)
|
df |
|
node |
name of top node in tree (to start descending from). |
current_str |
split condition to be passed through subsequent function calls in order to keep all conditions that lead to terminal nodes. Default is NULL. |
unique_expr |
should unique split expressions for terminal nodes be
returned? Default is |
Nested list
containing appeneded split conditions for terminal
nodes.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | 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)
tree_1_split_expr <- get_split_expressions(h2o_trees[[1]])
tree_1_terminal_nodes_list <- get_terminal_node_paths(df = tree_1_split_expr,
node = tree_1_split_expr[1, 'node'],
current_str = NULL)
|
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.