BERT regression & LIME explainer

Greetings,

I am looking to apply a LIME explainer to a fine-tuned BERT-model with a linear output layer. My training pipeline is vanilla, I am just stuck on integrating my model into the LIME explainer.

Training form of data in my training pipeline: List of sentences, each mapping onto a numeric value.

The idea is to use LIME to explain a BERT regression model.

If someone has solved this problem before or has an idea how, I’d be thankful if you let me know.

Did you find a solution for this?