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

Add show_values option to feature importances plot (#2351)

Browse Source 
Adding an option to remove the values from the features importances plot in Python.
This commit is contained in:
SimonAB 2017-08-31 18:26:54 +01:00 committed by Vadim Khotilovich
parent 0664298bb2
commit 2e9d06443e
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -15,7 +15,7 @@ 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',
importance_type='weight', max_num_features=None, importance_type='weight', max_num_features=None,
grid=True, **kwargs): grid=True, show_values=True, **kwargs):
"""Plot importance based on fitted trees. """Plot importance based on fitted trees.
@ -45,6 +45,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.
show_values : bool, default True
Show values on plot. To disable, pass False.
kwargs : kwargs :
Other keywords passed to ax.barh() Other keywords passed to ax.barh()
@ -83,8 +85,9 @@ def plot_importance(booster, ax=None, height=0.2,
ylocs = np.arange(len(values)) ylocs = np.arange(len(values))
ax.barh(ylocs, values, align='center', height=height, **kwargs) ax.barh(ylocs, values, align='center', height=height, **kwargs)
for x, y in zip(values, ylocs): if show_values is True:
ax.text(x + 1, y, x, va='center') for x, y in zip(values, ylocs):
ax.text(x + 1, y, x, va='center')
ax.set_yticks(ylocs) ax.set_yticks(ylocs)
ax.set_yticklabels(labels) ax.set_yticklabels(labels)