Add Pooling Average to Broken XXXForSentenceEmbedding annotators - Githubissues

JohnSnowLabs / spark-nlp

State of the Art Natural Language Processing

https://sparknlp.org/

Apache License 2.0

3.88k stars 711 forks source link

Add Pooling Average to Broken XXXForSentenceEmbedding annotators #14328

Closed ahmedlone127 closed 5 months ago

ahmedlone127 commented 5 months ago

This PR adds a pooling average to BertForSentenceEmbeddings and XlmRobertaForSentenceEmbeddings. It resolves the following issues: https://github.com/JohnSnowLabs/spark-nlp/issues/14181