File tree Expand file tree Collapse file tree 3 files changed +13
-4
lines changed
api/api_utils/assumptions Expand file tree Collapse file tree 3 files changed +13
-4
lines changed Original file line number Diff line number Diff line change @@ -93,7 +93,7 @@ def builders(self):
9393 return {
9494 'rfr' : PipelineBuilder ().add_node ('rfr' ),
9595 'ridge' : PipelineBuilder ().add_node ('ridge' ),
96- 'lgbm ' : PipelineBuilder ().add_node ('lgbm ' )
96+ 'lgbmreg ' : PipelineBuilder ().add_node ('lgbmreg ' )
9797 }
9898
9999 def ensemble_operation (self ) -> str :
Original file line number Diff line number Diff line change 1111 "booster" : " gbtree" ,
1212 "tree_method" : " auto" ,
1313 "enable_categorical" : true ,
14- "use_eval_set" : true
14+ "use_eval_set" : true ,
15+ "early_stopping_rounds" : 30
1516 },
1617 "xgboostreg" : {
1718 "n_jobs" : 1 ,
1819 "verbosity" : 0 ,
1920 "booster" : " gbtree" ,
2021 "tree_method" : " auto" ,
2122 "enable_categorical" : true ,
22- "use_eval_set" : true
23+ "use_eval_set" : true ,
24+ "early_stopping_rounds" : 30
2325 },
2426 "catboost" : {
2527 "n_jobs" : -1 ,
4042 "nan_mode" : " Min" ,
4143 "verbose" : false ,
4244 "allow_writing_files" : false ,
45+ "use_eval_set" : true ,
46+ "use_best_model" : true ,
4347 "enable_categorical" : true
4448 },
4549 "catboostreg" : {
6165 "nan_mode" : " Min" ,
6266 "verbose" : false ,
6367 "allow_writing_files" : false ,
68+ "use_eval_set" : true ,
69+ "use_best_model" : true ,
6470 "enable_categorical" : true
6571 },
6672 "lgbm" : {
6975 "bagging_fraction" : 0.85 ,
7076 "extra_trees" : false ,
7177 "enable_categorical" : true ,
78+ "use_eval_set" : true ,
79+ "early_stopping_rounds" : 30 ,
7280 "n_jobs" : 1 ,
7381 "verbose" : -1
7482 },
7886 "bagging_fraction" : 0.85 ,
7987 "extra_trees" : false ,
8088 "enable_categorical" : true ,
89+ "use_eval_set" : true ,
90+ "early_stopping_rounds" : 30 ,
8191 "n_jobs" : 1 ,
8292 "verbose" : -1
8393 },
Original file line number Diff line number Diff line change @@ -26,7 +26,6 @@ sktime>=0.16.1; python_version >= '3.10'
2626hyperopt == 0.2.7
2727SALib >= 1.3.0
2828scikit-optimize >= 0.7.4
29- optuna >= 3.2.0
3029
3130# Plotting
3231matplotlib >= 3.3.1 ; python_version >= '3.8'
You can’t perform that action at this time.
0 commit comments