keras-io
/

TF_Decision_Trees

Tabular Classification

Model card Files Files and versions Metrics Training metrics Community

tdubon commited on Feb 15, 2022

Commit

7e80847

•

1 Parent(s): c9c1944

Update config.json

Files changed (1) hide show

config.json +10 -10

config.json CHANGED Viewed

@@ -1,15 +1,15 @@
 {
-    "architectures": [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"
 }

 {
+    "architectures": "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"
 }