Merge d72fb0b into 4b7e0e7

ClimbsRocks · Feb 10, 2021 · f9331a3 · f9331a3
2 parents 4b7e0e7 + d72fb0b
commit f9331a3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/auto_ml/utils_models.py b/auto_ml/utils_models.py
@@ -173,8 +173,8 @@ def get_model_from_name(model_name, training_params=None, is_hp_search=False):
         model_map['LGBMClassifier'] = LGBMClassifier()
 
     if catboost_installed:
-        model_map['CatBoostRegressor'] = CatBoostRegressor(calc_feature_importance=True)
-        model_map['CatBoostClassifier'] = CatBoostClassifier(calc_feature_importance=True)
+        model_map['CatBoostRegressor'] = CatBoostRegressor()
+        model_map['CatBoostClassifier'] = CatBoostClassifier()
 
     if model_name[:12] == 'DeepLearning':
         if keras_imported == False: