Skip to content

Commit c2fac18

Browse files
committed
Update AutoGluon params
1 parent 90a55c6 commit c2fac18

File tree

1 file changed

+12
-9
lines changed

1 file changed

+12
-9
lines changed

Diff for: examples/ames_housing/configs/4_ames_housing_autogluon.json

+12-9
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
"task": "regression",
99
"tracking": {
1010
"experiment": "ames_housing",
11-
"run": "baseline"
11+
"run": "AutoGluon"
1212
}
1313
},
1414
"steps": [
@@ -19,15 +19,16 @@
1919
"predict_path": "examples/ames_housing/data/test.csv",
2020
"drop_columns": [
2121
"Id"
22-
]
22+
],
23+
"optimize_dtypes": true
2324
}
2425
},
2526
{
2627
"step_type": "TabularSplitStep",
2728
"parameters": {
28-
"train_percentage": 0.7,
29+
"train_percentage": 0.6,
2930
"validation_percentage": 0.2,
30-
"test_percentage": 0.1
31+
"test_percentage": 0.2
3132
}
3233
},
3334
{
@@ -39,15 +40,17 @@
3940
{
4041
"step_type": "AutoGluonModelStep",
4142
"parameters": {
42-
"model_class": "AutoGluonModel",
43-
"autogluon_create_params": {},
43+
"model_class": "AutoGluon",
44+
"autogluon_create_params": {
45+
"verbosity": 2
46+
},
4447
"autogluon_fit_params": {
4548
"presets": [
46-
"high_quality"
49+
"high_quality",
50+
"optimize_for_deployment"
4751
],
52+
"save_bag_folds": true,
4853
"time_limit": 1800,
49-
"num_bag_folds": 5,
50-
"num_bag_sets": 2,
5154
"num_stack_levels": 1,
5255
"dynamic_stacking": false
5356
}

0 commit comments

Comments
 (0)