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

Allow pass fmap to importance plot (#5719)

Browse Source 
Co-authored-by: Peter Jung <peter.jung@heureka.cz>
Co-authored-by: Hyunsu Cho <chohyu01@cs.washington.edu>
This commit is contained in:
Peter Jung 2020-05-29 13:55:35 +02:00 committed by GitHub
parent f779980f7e
commit 251dc8a663
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
python-package/xgboost/plotting.py
View File

@ -10,7 +10,7 @@ from .sklearn import XGBModel
def plot_importance(booster, ax=None, height=0.2, def plot_importance(booster, ax=None, height=0.2,
xlim=None, ylim=None, title='Feature importance', xlim=None, ylim=None, title='Feature importance',
xlabel='F score', ylabel='Features', xlabel='F score', ylabel='Features', fmap='',
importance_type='weight', max_num_features=None, importance_type='weight', max_num_features=None,
grid=True, show_values=True, **kwargs): grid=True, show_values=True, **kwargs):
"""Plot importance based on fitted trees. """Plot importance based on fitted trees.
@ -43,6 +43,8 @@ def plot_importance(booster, ax=None, height=0.2,
X axis title label. To disable, pass None. X axis title label. To disable, pass None.
ylabel : str, default "Features" ylabel : str, default "Features"
Y axis title label. To disable, pass None. Y axis title label. To disable, pass None.
fmap: str or os.PathLike (optional)
The name of feature map file.
show_values : bool, default True show_values : bool, default True
Show values on plot. To disable, pass False. Show values on plot. To disable, pass False.
kwargs : kwargs :
@ -59,9 +61,9 @@ def plot_importance(booster, ax=None, height=0.2,
if isinstance(booster, XGBModel): if isinstance(booster, XGBModel):
importance = booster.get_booster().get_score( importance = booster.get_booster().get_score(
importance_type=importance_type) importance_type=importance_type, fmap=fmap)
elif isinstance(booster, Booster): elif isinstance(booster, Booster):
importance = booster.get_score(importance_type=importance_type) importance = booster.get_score(importance_type=importance_type, fmap=fmap)
elif isinstance(booster, dict): elif isinstance(booster, dict):
importance = booster importance = booster
else: else: