5

我正在尝试在 Python 中绘制一些SHAP 图,以更深入地了解我的机器学习模型的输出。这是我在 for 循环中调用的方法:

def plotAndSaveSHAPSummary(model,train_data,x_train,pathToSHAPPlots):
    shap_values = model.get_feature_importance(train_data, type='ShapValues')
    expected_value = shap_values[0,-1]
    shap_values = shap_values[:,:-1]

    shap.summary_plot(shap_values,x_train,max_display=20,show=False)
    plt.savefig(pathToSHAPPlots+'/SHAP Plots/SHAP_Plot'+str(counter)+'.png',dpi=300,bbox_inches='tight')
    plt.clf()

绘图按预期保存到磁盘,但在每次调用 savefig 方法后,我收到以下错误消息:

Exception in Tkinter callback
Traceback (most recent call last):
  File "D:\PathTo\Anaconda\Lib\tkinter\__init__.py", line 1705, in __call__
    return self.func(*args)
  File "D:\PathTo\Anaconda\Lib\tkinter\__init__.py", line 749, in callit
    func(*args)
  File "D:\PathTo\Anaconda\lib\site-packages\matplotlib\backends\_backend_tk.py", line 270, in idle_draw
    self.draw()
  File "D:\PathTo\Anaconda\lib\site-packages\matplotlib\backends\backend_tkagg.py", line 9, in draw
    super(FigureCanvasTkAgg, self).draw()
  File "D:\PathTo\Anaconda\lib\site-packages\matplotlib\backends\backend_agg.py", line 393, in draw
    self.figure.draw(self.renderer)
  File "D:\PathTo\Anaconda\lib\site-packages\matplotlib\backend_bases.py", line 1535, in _draw
    def _draw(renderer): raise Done(renderer)
matplotlib.backend_bases._get_renderer.<locals>.Done: <matplotlib.backends.backend_agg.RendererAgg object at 0x000002066B288288>

任何想法如何摆脱这个异常?

4

4 回答 4

4

在更新其中一个库(可能是 matplot 库)后,我也开始收到此错误。评论中描述的使用建议plt.pause有效,但它是内部调用plt.show(),这对我来说是不可接受的。我最终canvas.start_event_loopsavefig

import sys

def _save(self, filename: str) -> None:
   fig1 = plt.gcf()
   fig1.set_size_inches(6, 2.8)
   plt.draw()
   fig1.subplots_adjust(left=0.20)
   fig1.canvas.start_event_loop(sys.float_info.min) #workaround for Exception in Tkinter callback
   fig1.savefig(self.dst_dir + '/' + filename + '.png', dpi=220, bbox_inches='tight')
   fig1.clf()
   plt.close()
于 2020-09-30T15:25:31.703 回答
1
plt.ioff()

为我做了伎俩

于 2021-01-01T17:02:51.540 回答
0

我有同样的问题。我在 Python 中有一个简单的脚本来制作一个过去可以正常工作的图表,但现在当它尝试保存文件时,会显示确切的警告。我花了很多时间试图弄清楚发生了什么,因为错误是突然的,而且信息不是很丰富。

我发现当我省略bbox_inches参数 at时它消失了savefig

我注意到安装后错误开始发生python-tk,但卸载它并没有帮助。

[Mariu 的回答]plt.ioff()对我不起作用。

[nan的回答]fig1.canvas.start_event_loop(sys.float_info.min)确实有效。

于 2021-01-05T23:55:24.877 回答
0

使用 matplot lib 保存多个图时,我遇到了同样的问题。保存数字后添加 plt.close() 对我有用。

于 2021-03-12T13:25:50.977 回答