Update config.json

config.json

@@ -1,14 +1,15 @@
-{
-    "model": ["Tfdf.keras.GradientBoostedTreesModel"],
-    "growing_strategy": "BEST_FIRST_GLOBAL",
-    "num_trees": "250",
-    "min_examples": "6",
-    "max_depth": "5",
-    "subsample": "0.65",
-    "validation_ratio": "0.1",
-    "sampling_method": "RANDOM",
-    "batch_size": "None",
-    "num_epochs": "1",
-    "task": "tfdf.keras.Task.CLASSIFICATION",
-    "loss": "DEFAULT"
+
+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 {} }
+ }
 }
+