积分梯度错误 - 尝试将具有不受支持的类型 (<class 'NoneType'>) 的值 (None) 转换为张量

Integrated Gradients Error - Attempt to convert a value (None) with an unsupported type (<class 'NoneType'>) to a Tensor

我是 transformers 的新手,但我设法使用 tensorflow 创建了一个 Bert 分类器,现在我想实现 Integreted Gradients 来解释模型,但是我得到这个错误:

Attempt to convert a value (None) with an unsupported type (<class 'NoneType'>) to a Tensor

如果您认为可以提供帮助,请在此处找到我的代码:link 因为要粘贴到此处的代码很多。

在您的笔记本中,错误是由

触发的
explanation = ig.explain(a,
                         baselines=None,
                         target=predictions)

因此,似乎ig需要非空基线。