Lime explanation does not match the number of classes

marcotcr / lime

Lime: Explaining the predictions of any machine learning classifier

BSD 2-Clause "Simplified" License

11.5k stars 1.79k forks source link

Lime explanation does not match the number of classes #588

Open skocbek opened 3 years ago

skocbek commented 3 years ago

Hi, I have built a text classification model that classifies input into two classes. When I apply Lime to the model, the output contains prediction probabilities for more than two classes. In addition, probabilities for all classes are 1.0. What went wrong?

marcotcr commented 3 years ago

what's the output of your function for x=['john is a man', 'mary is a woman']?