python - 如何将 xgboost 的特征重要性图从 Jupyter 笔记本保存到文件

Question

我正在努力将 xgboost 特征重要性图保存到文件中。我在我的 jupyter notebook 中创建了一个模型并绘制了功能的重要性 -

xgb_model = xgboost.train(best_params, dtrain, num_round)
xgboost.plot_importance(xgb_model)

它向我显示了特征重要性图，但我无法将其保存到文件中。我什至找过任何 保存 dir(xgboost.plot_importance(xgb_model)) 中的属性，却一无所获。有没有办法做到这一点？

Answer 1

根据 doc , xgboost.plot_importance(xgb_model)返回 matplotlib Axes
因此，你可以

ax = xgboost.plot_importance(xgb_model)
ax.figure.savefig('the-path-you-want-to-save.png')

另外，如果您丢失了图形的左右边距，您可以设置 tight_layout

ax = xgboost.plot_importance(xgb_model)
ax.figure.tight_layout()
ax.figure.savefig('the-path-you-want-to-save.png')