tdubon commited on
Commit
9f87a4e
1 Parent(s): 3cf7290

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +14 -13
config.json CHANGED
@@ -1,14 +1,15 @@
1
- {
2
- "model": ["Tfdf.keras.GradientBoostedTreesModel"],
3
- "growing_strategy": "BEST_FIRST_GLOBAL",
4
- "num_trees": "250",
5
- "min_examples": "6",
6
- "max_depth": "5",
7
- "subsample": "0.65",
8
- "validation_ratio": "0.1",
9
- "sampling_method": "RANDOM",
10
- "batch_size": "None",
11
- "num_epochs": "1",
12
- "task": "tfdf.keras.Task.CLASSIFICATION",
13
- "loss": "DEFAULT"
14
  }
 
 
1
+
2
+ learner: "GRADIENT_BOOSTED_TREES"
3
+
4
+ [yggdrasil_decision_forests.model.gradient_boosted_trees.proto.gradient_boosted_trees_config] {
5
+ num_trees: 1000
6
+ use_hessian_gain: true
7
+ validation_set_ratio: 0.0
8
+ early_stopping: NONE
9
+ decision_tree {
10
+ growing_strategy_best_first_global { max_num_nodes: 64 }
11
+ sparse_oblique_split {}
12
+ categorical { random {} }
13
+ }
14
  }
15
+