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

Asked By: dsl1990

||

Answer 1

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 By: tacaswell

Answer 2

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

Answered By: yanachen

Question: