Update pipeline.py
Browse files- pipeline.py +2 -1
pipeline.py
CHANGED
@@ -659,7 +659,8 @@ class Pipeline(object):
|
|
659 |
return predictions,embeddings_output
|
660 |
|
661 |
def bert_interpretability(self, input_text:str):
|
662 |
-
|
|
|
663 |
|
664 |
|
665 |
if __name__ == '__main__':
|
|
|
659 |
return predictions,embeddings_output
|
660 |
|
661 |
def bert_interpretability(self, input_text:str):
|
662 |
+
clean_data,_ = self.bert_model.clean_data(input_text)
|
663 |
+
return self.lime_interpretability.lime_analysis(self.bert_model,input_text, clean_data, class_names=self.bert_model.config['classes'])
|
664 |
|
665 |
|
666 |
if __name__ == '__main__':
|