encode_terminal_nodes: Encode input data as terminal nodes in tree based model.

Description Usage Arguments Value Examples

Description

Encode input data as terminal nodes in tree based model.

Usage

1
2
encode_terminal_nodes(data, terminal_node_split_rules, h2o_trees_in_R,
  return_one_hot = TRUE)

Arguments

data

input data to encode as terminal nodes from trees. Currently only data.frame inputs are supported.

terminal_node_split_rules

list of splitting rules for terminal nodes in each tree, the output from extract_split_rules function.

h2o_trees_in_R

list of trees output from H2OTreeConvertR::h2o_tree_convertR function.

Value

terminal node encoding for input data. Currently only data.frame input data is supported.

Examples

 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 <- encode_terminal_nodes(data = prostate_df,
                                                terminal_node_split_rules = h2o_trees_terminal_node_rules,
                                                h2o_trees_in_R = h2o_trees)

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