使用 SHAP 的 Tkinter 回调异常

Posted 2023-02-16

技术标签:

【中文标题】使用 SHAP 的 Tkinter 回调异常【英文标题】：Exception in Tkinter callback using SHAP 【发布时间】：2020-09-16 22:56:25 【问题描述】：

我正在尝试在 Python 中绘制一些 SHAP plots，以更深入地了解我的机器学习模型的输出。这是我在 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>

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

【问题讨论】：

我最近只使用 matplotlib 遇到了同样的错误，并通过在 savefig 调用之前添加一个小的 plt.pause(1e-13) 来解决它（hackily）。我不知道为什么会这样，但假设 matplotlib 代码中的某处存在竞争条件。 【参考方案1】：

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

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()

【讨论】：

【参考方案2】：