如何在xgboost.plot_importance中更改绘图的大小？

Question

xgboost.plot_importance(model, importance_type='gain')

我无法改变这个情节的大小。我想以适当的大小保存这个数字，以便我可以在pdf中使用它。我希望类似figize

Answer 1

看起来像plot_importance return an Axes object

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

看起来你也可以在

中传递轴

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