TF_Decision_Trees / config.json
tdubon's picture
Update config.json
9f87a4e
learner: "GRADIENT_BOOSTED_TREES"
[yggdrasil_decision_forests.model.gradient_boosted_trees.proto.gradient_boosted_trees_config] {
num_trees: 1000
use_hessian_gain: true
validation_set_ratio: 0.0
early_stopping: NONE
decision_tree {
growing_strategy_best_first_global { max_num_nodes: 64 }
sparse_oblique_split {}
categorical { random {} }
}
}