From 0080dfc7c8375722ebfd836ccf56aabfc4077b27 Mon Sep 17 00:00:00 2001 From: ClimbsRocks Date: Wed, 29 Nov 2017 08:09:10 -0800 Subject: [PATCH] more error handling for xgb's eccentricities --- auto_ml/predictor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/auto_ml/predictor.py b/auto_ml/predictor.py index 111f02e..cb4937c 100644 --- 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: