huggingface / neuralcoref

✨Fast Coreference Resolution in spaCy with Neural Networks
https://huggingface.co/coref/
MIT License
2.83k stars 474 forks source link

Add doc embedding implementation during inference #339

Open valedica opened 2 years ago

valedica commented 2 years ago

This closes #338. The implementation follows the one of the method get_document_embedding from neuralcoref.train.document, which is the method that calculates document embeddings during training.