Pythonで作成したLightGBMの出力結果を画像データとして保存したい

Question

PythonでLightGBMを用いて「feature importance」と「決定木」をJupyter Notebook上で出力しました。
ここに出力したものを画像データとして保存し、他の人と共有したいと考えています。
以下のようなコードにより「feature importance」と「決定木」を出力したのですが、どのようにすれば画像データを保存することができるでしょうか。

gbm=lgb.train(params,train_data)

「feature importance」←この画像を保存したい
lgb.plot_importance(gbm,height=0.5,figsize=(8,16))

「決定木」←この画像を保存したい
lgb.create_tree_digraph(gb

Answer 1

試してませんが

gbm=lgb.train(params,train_data)
lgb.plot_importance(gbm,height=0.5,figsize=(8,16))
plt.savefig('importance.png')

lgb.create_tree_digraph(gb
plt.savefig('tree_digraph.png')

で保存できると思います。一度お試しください。ファイル名は適宜変更してください。

詳しいパラメータはこちら。
https://matplotlib.org/stable/api/_as_gen/matplotlib.pyplot.savefig.html

Comments

[trave]

ありがとうございます。
解決しました。