{ | |
"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" | |
} | |