huggingface / neuralcoref

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

Scores between the mentions #92

Closed sanjanalreddy closed 5 years ago

sanjanalreddy commented 6 years ago

I am trying to get the scores which are shown on the neuralcoref viz but don't seem to be able to do so. I tried to call doc._.coref_scores but it mentions that the extension has not been set. Has anyone been able to get the scores?

devitoclic commented 6 years ago

I'm having the same issue right now. I saw doc._.coref_scores in the source code from 3.0.0 but i cannot use it since it's not released as a spacy package yet. Do you have any news about this feature ?

sanjanalreddy commented 6 years ago

Looked a little more into this issue. Looks like the code to get the coreference scores is there, but the models have to be retrained. So one of the solutions is to pull the latest version of master and then follow the instructions to retrain the models. Doing so, will enable the extension doc._.coref_scores.

thomwolf commented 5 years ago

The latest release (4.0) now outputs coreference scores.