File tree 1 file changed +12
-9
lines changed
examples/ames_housing/configs
1 file changed +12
-9
lines changed Original file line number Diff line number Diff line change 8
8
"task" : " regression" ,
9
9
"tracking" : {
10
10
"experiment" : " ames_housing" ,
11
- "run" : " baseline "
11
+ "run" : " AutoGluon "
12
12
}
13
13
},
14
14
"steps" : [
19
19
"predict_path" : " examples/ames_housing/data/test.csv" ,
20
20
"drop_columns" : [
21
21
" Id"
22
- ]
22
+ ],
23
+ "optimize_dtypes" : true
23
24
}
24
25
},
25
26
{
26
27
"step_type" : " TabularSplitStep" ,
27
28
"parameters" : {
28
- "train_percentage" : 0.7 ,
29
+ "train_percentage" : 0.6 ,
29
30
"validation_percentage" : 0.2 ,
30
- "test_percentage" : 0.1
31
+ "test_percentage" : 0.2
31
32
}
32
33
},
33
34
{
39
40
{
40
41
"step_type" : " AutoGluonModelStep" ,
41
42
"parameters" : {
42
- "model_class" : " AutoGluonModel" ,
43
- "autogluon_create_params" : {},
43
+ "model_class" : " AutoGluon" ,
44
+ "autogluon_create_params" : {
45
+ "verbosity" : 2
46
+ },
44
47
"autogluon_fit_params" : {
45
48
"presets" : [
46
- " high_quality"
49
+ " high_quality" ,
50
+ " optimize_for_deployment"
47
51
],
52
+ "save_bag_folds" : true ,
48
53
"time_limit" : 1800 ,
49
- "num_bag_folds" : 5 ,
50
- "num_bag_sets" : 2 ,
51
54
"num_stack_levels" : 1 ,
52
55
"dynamic_stacking" : false
53
56
}
You can’t perform that action at this time.
0 commit comments