Related text to the predicted entity

#4
by ElKe5a - opened

Hi @sadickam ,

Thank you very much for your work and for making this model available! I wanted to know if it was possible to also extract the text that references the predicted entity.

Thanks,

Hello @ElKe5a

It is possible to extract the text reference for the prediction. You can do that using the SHAP python package. Please have a look at the SHAP documentation. It has a specific function for explaining the predictions of NLP models with very helpful visualisations.

Hopefully, this helps. If you need further information please do let me know.

Regards
Sadick

Sign up or log in to comment