diff --git a/fedot/core/operations/evaluation/boostings.py b/fedot/core/operations/evaluation/boostings.py index bdffa1fd6f..1e359d787d 100644 --- a/fedot/core/operations/evaluation/boostings.py +++ b/fedot/core/operations/evaluation/boostings.py @@ -16,7 +16,7 @@ class BoostingStrategy(EvaluationStrategy): 'catboost': FedotCatBoostClassificationImplementation, 'catboostreg': FedotCatBoostRegressionImplementation, 'xgboost': FedotXGBoostClassificationImplementation, - 'xgboostreg': FedotXGBoostRegressionImplementation + 'xgboostreg': FedotXGBoostRegressionImplementation, 'lgbm': FedotLightGBMClassificationImplementation, 'lgbmreg': FedotLightGBMRegressionImplementation }