Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Check booster for dart in feature importance. (#4073)

Browse Source 
* Check booster for dart in feature importance.
This commit is contained in:
Jiaming Yuan 2019-01-22 16:03:54 +08:00 committed by GitHub
parent 301cef4638
commit 4fac9874e0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
python-package/xgboost/core.py
View File

@ -1448,8 +1448,7 @@ class Booster(object):
importance_type: str, default 'weight'
One of the importance types defined above.
"""
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 ValueError('Feature importance is not defined for Booster type {}'
.format(self.booster))