Description Usage Arguments Value Examples
This function returns the results as a data.frame.
1 2 | encode_terminal_nodes_df(data, terminal_node_split_rules, h2o_tree_in_R,
tree_name = "tree_", return_one_hot = TRUE)
|
data |
input data to encode as terminal nodes from trees. Currently only |
terminal_node_split_rules |
splitting rules for terminal nodes in a single tree,
the output from |
h2o_tree_in_R |
single tree output from |
tree_name |
the prefix to be given to column names. Columns will be named prefix followed by the name of the terminal node. Default = 'tree_'. |
terminal node encoding for input data. Currently only data.frame
is output.
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)
h2o_trees_terminal_node_rules <- extract_split_rules(h2o_trees)
terminal_node_encoding_tree1 <- encode_terminal_nodes(data = prostate_df,
terminal_node_split_rules = h2o_trees_terminal_node_rules[[1]],
h2o_trees_in_R = h2o_trees[[1]])
|
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.