more error handling for xgb's eccentricities

ClimbsRocks · Nov 29, 2017 · 0080dfc · 0080dfc
1 parent 766fa0f
commit 0080dfc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/auto_ml/predictor.py b/auto_ml/predictor.py
@@ -1003,7 +1003,7 @@ def print_results(self, model_name, model, X, y):
 
                 feature_responses = self.create_feature_responses(model, X, y, top_features)
                 self._join_and_print_analytics_results(feature_responses, sorted_model_results, sort_field='Importance')
-            except TypeError as e:
+            except AttributeError as e:
                 if model_name == 'XGBRegressor':
                     pass
                 else: