How to change size of plot in xgboost.plot_importance?

Question

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

Solution

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)

Comments

[Max Power]

^ only the second option works for me as well. Regardless, thanks for the answer!

[tbrodbeck]

it is first width and then height btw :)