How to change size of plot in xgboost.plot_importance?

dsl1990 picture dsl1990 · Nov 17, 2016 · Viewed 12.4k times · Source
xgboost.plot_importance(model, importance_type='gain')

I am not able to change size of this plot. I want to save this figure with proper size so that I can use it in pdf. I want similar like figize

Answer

tacaswell picture tacaswell · Nov 18, 2016

It looks like plot_importance return an Axes object

ax = xgboost.plot_importance(...)
fig = ax.figure
fig.set_size_inches(h, w)

It also looks like you can pass an axes in

fig, ax = plt.subplots(figsize=(h, w))
xgboost.plot_importance(..., ax=ax)