Hiding unnecessary code

aimclub · Sep 6, 2023 · c3fb410 · c3fb410
1 parent 4ef568b
commit c3fb410
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 8 deletions.
diff --git a/fedot/core/operations/evaluation/boostings.py b/fedot/core/operations/evaluation/boostings.py
@@ -10,9 +10,7 @@
 
 class BoostingClassificationStrategy(EvaluationStrategy):
     __operations_by_types = {
-        'catboost': FedotCatBoostClassificationImplementation,
-        # 'xgboost': FedotXgboostBoostClassificationImplementation,
-        # 'lgbm': FedotLightGBMBoostClassificationImplementation,
+        'catboost': FedotCatBoostClassificationImplementation
     }
 
     def __init__(self, operation_type: str, params: Optional[OperationParameters] = None):
@@ -49,9 +47,7 @@ def predict(self, trained_operation, predict_data: InputData) -> OutputData:
 
 class BoostingRegressionStrategy(EvaluationStrategy):
     __operations_by_types = {
-        'catboostreg': FedotCatBoostRegressionImplementation,
-        # 'xgb_regr': FedotXgboostBoostRegressionImplementation,
-        # 'lgbm_regr': FedotLightGBMBoostRegressionImplementation,
+        'catboostreg': FedotCatBoostRegressionImplementation
     }
 
     def __init__(self, operation_type: str, params: Optional[OperationParameters] = None):

diff --git a/fedot/core/operations/evaluation/evaluation_interfaces.py b/fedot/core/operations/evaluation/evaluation_interfaces.py
@@ -182,7 +182,6 @@ class SkLearnEvaluationStrategy(EvaluationStrategy):
         'svr': SklearnSVR,
         'sgdr': SklearnSGD,
         'lgbmreg': LGBMRegressor,
-        # 'catboostreg': CatBoostRegressor,
 
         'xgboost': XGBClassifier,
         'logit': SklearnLogReg,
@@ -192,7 +191,6 @@ class SkLearnEvaluationStrategy(EvaluationStrategy):
         'rf': RandomForestClassifier,
         'mlp': MLPClassifier,
         'lgbm': LGBMClassifier,
-        # 'catboost': CatBoostClassifier,
 
         'kmeans': SklearnKmeans,
     }