JohnSnowLabs / spark-nlp

State of the Art Natural Language Processing
https://sparknlp.org/
Apache License 2.0
3.76k stars 703 forks source link

Add Pooling Average to Broken XXXForSentenceEmbedding annotators #14328

Closed ahmedlone127 closed 3 weeks ago

ahmedlone127 commented 3 weeks ago

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