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

5
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',
xlabel='F score', ylabel='Features',
importance_type='weight', max_num_features=None,
grid=True, **kwargs):
grid=True, show_values=True, **kwargs):
"""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.
ylabel : str, default "Features"
Y axis title label. To disable, pass None.
show_values : bool, default True
Show values on plot. To disable, pass False.
kwargs :
Other keywords passed to ax.barh()
@ -83,6 +85,7 @@ def plot_importance(booster, ax=None, height=0.2,
ylocs = np.arange(len(values))
ax.barh(ylocs, values, align='center', height=height, **kwargs)
if show_values is True:
for x, y in zip(values, ylocs):
ax.text(x + 1, y, x, va='center')