Hugging Face's logo

sadickam
/
sdgBERT

Text Classification
Transformers
Safetensors
English
bert
Inference Endpoints
Model card Files Community
6

Related text to the predicted entity

#4
by ElKe5a - opened
Discussion
ElKe5a

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,

sadickam
 Owner

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

sadickam changed discussion status to closed

Sign up or log in to comment