用SHAP做蜂群图后想保存内容的保存代码

用模型跑完结果后得到shap的蜂群图如下所示

代码：

shap.summary_plot(shap_values)

显示结果

想要将上述的蜂群图进行保存，可以使用如下的代码：

import matplotlib.pyplot as plt
import shap

# 生成 SHAP 值（假设你已经有了 shap_values）
# shap_values = ...

# 生成 shap.summary_plot，但不显示它
shap.summary_plot(shap_values, show=False)

# 获取当前图像对象
fig = plt.gcf()

# 保存图像到指定文件夹，设置 dpi 为 400
fig.savefig('path/to/your/folder/shap_summary_plot.jpg', dpi=400)

# 如果你想显示图片
plt.show()

一定注意上述代码要同时运行，不然会保存为空白图片。