Skip to content

Commit 51ecb5f

Browse files
committed
fix: small fixes
1 parent 096f830 commit 51ecb5f

File tree

3 files changed

+13
-4
lines changed

3 files changed

+13
-4
lines changed

fedot/api/api_utils/assumptions/task_assumptions.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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:

fedot/core/repository/data/default_operation_params.json

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,15 +11,17 @@
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,
@@ -40,6 +42,8 @@
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": {
@@ -61,6 +65,8 @@
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": {
@@ -69,6 +75,8 @@
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
},
@@ -78,6 +86,8 @@
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
},

requirements.txt

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,6 @@ sktime>=0.16.1; python_version >= '3.10'
2626
hyperopt==0.2.7
2727
SALib>=1.3.0
2828
scikit-optimize>=0.7.4
29-
optuna>=3.2.0
3029

3130
# Plotting
3231
matplotlib>=3.3.1; python_version >= '3.8'

0 commit comments

Comments
 (0)