TF_Decision_Trees / config.json
tdubon's picture
Upload config.json
c9c1944
raw
history blame
369 Bytes
{
"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"
}