TF_Decision_Trees / config.json
tdubon's picture
Update config.json
642a884
raw
history blame
381 Bytes
{
"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"
}