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

fix ylim with max_num_features in python plot_importance (#1974)

Browse Source
This commit is contained in:
wxchan 2017-01-19 03:59:50 +08:00 committed by Tianqi Chen
parent a7d2833766
commit a073a2c3d4
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/plotting.py
View File

@ -73,7 +73,6 @@ def plot_importance(booster, ax=None, height=0.2,
tuples = [(k, importance[k]) for k in importance] tuples = [(k, importance[k]) for k in importance]
if max_num_features is not None: if max_num_features is not None:
tuples = sorted(tuples, key=lambda x: x[1])[-max_num_features:] tuples = sorted(tuples, key=lambda x: x[1])[-max_num_features:]
ylim = (-1, max_num_features)
else: else:
tuples = sorted(tuples, key=lambda x: x[1]) tuples = sorted(tuples, key=lambda x: x[1])
labels, values = zip(*tuples) labels, values = zip(*tuples)
@ -101,7 +100,7 @@ def plot_importance(booster, ax=None, height=0.2,
if not isinstance(ylim, tuple) or len(ylim) != 2: if not isinstance(ylim, tuple) or len(ylim) != 2:
raise ValueError('ylim must be a tuple of 2 elements') raise ValueError('ylim must be a tuple of 2 elements')
else: else:
ylim = (-1, len(importance)) ylim = (-1, len(values))
ax.set_ylim(ylim) ax.set_ylim(ylim)
if title is not None: if title is not None: