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

score 49 · Accepted Answer · answered Nov 18 '16 at 04:00

49

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)

answered Nov 18 '16 at 04:00

tacaswell

1

^ only the second option works for me as well. Regardless, thanks for the answer! – Max Power Jan 16 '18 at 15:33
1

it is first width and then height btw :) – tbrodbeck Jul 14 '21 at 12:53

score 1 · Answer 2 · answered Nov 07 '22 at 15:51

1

import matplotlib.pyplot as plt
fig, ax = plt.subplots(figsize=(10, 20))
xgb.plot_importance(bst, ax=ax)
plt.show()

answered Nov 07 '22 at 15:51

yanachen

2 Answers2