From c2a3902ba3cf3263c733ef9ed686b24ee9223e8b Mon Sep 17 00:00:00 2001 From: Philip Hyunsu Cho Date: Fri, 31 May 2019 15:11:57 -0700 Subject: [PATCH] Fix #4497: Enable feature importance property for DART booster (#4525) --- python-package/xgboost/sklearn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python-package/xgboost/sklearn.py b/python-package/xgboost/sklearn.py index ae4751ea74e8..b80a79fcffa7 100644 --- a/python-package/xgboost/sklearn.py +++ b/python-package/xgboost/sklearn.py @@ -536,7 +536,7 @@ def feature_importances_(self): feature_importances_ : array of shape ``[n_features]`` """ - if getattr(self, 'booster', None) is not None and self.booster != 'gbtree': + if getattr(self, 'booster', None) is not None and self.booster not in {'gbtree', 'dart'}: raise AttributeError('Feature importance is not defined for Booster type {}' .format(self.booster)) b = self.get_booster()