matplotlib 如何在xgboost.plot_importance中改变绘图大小？

ygya80vv 于 2022-11-15 发布在其他

关注(0)|答案(2)|浏览(150)

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

我不能改变这个图的大小。我想保存这个图与适当的大小，以便我可以使用它在pdf。我想类似的像figize

2条答案

看起来plot_importance返回了一个Axes对象

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)

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