From d72fb0b53394ed41e0a646303fe1ea532d064235 Mon Sep 17 00:00:00 2001
From: elnurmmmdv <60216620+elnurmmmdv@users.noreply.github.com>
Date: Wed, 10 Feb 2021 11:43:03 +0400
Subject: [PATCH] Update utils_models.py

---
 auto_ml/utils_models.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/auto_ml/utils_models.py b/auto_ml/utils_models.py
index 4fb5898..187d18c 100644
--- 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: