cdpierse / transformers-interpret

Model explainability that works seamlessly with 🤗 transformers. Explain your transformers model in just 2 lines of code.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to use classification explainer for custom models?

Hossein-1991 opened this issue · comments

I've just been introduced to your package (it is so helpful).
I have a question regarding the classification explainer. I've fine-tuned a distilbert model on a specific dataset with my own designed classifier (several fc layers) on top of that (so, my model is not in the pretrained bert models):
soal

when I try to use the package to interpret outcomes, it gives this error:
'Distilbert' object has no attribute 'config'.
How should I fix that?
Thanks