TF-GB-Forest / config.json
tdubon's picture
Update config.json
a4d56bf
raw
history blame contribute delete
No virus
390 Bytes
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.1
early_stopping: NONE
GradientBoostedTreesModel {
growing_strategy_best_first_global { max_num_nodes: 64 }
sparse_oblique_split {}
categorical { random {} }
get_compile_flags()
}
}