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 {} } | |
} | |
} | |