Coreference resolution for multiple languages

Coreferee is a pipeline plugin that performs coreference resolution for English, German and Polish. It is designed so that it is easy to add support for new languages and optimised for limited training data. It uses a mixture of neural networks and programmed rules. Please note you will need to install models before running the code example.


import coreferee, spacy nlp = spacy.load('en_core_web_trf') nlp.add_pipe('coreferee') doc = nlp('Although he was very busy with his work, Peter had had enough of it. He and his wife decided they needed a holiday. They travelled to Spain because they loved the country very much.') doc._.coref_chains.print() # Output: # # 0: he(1), his(6), Peter(9), He(16), his(18) # 1: work(7), it(14) # 2: [He(16); wife(19)], they(21), They(26), they(31) # 3: Spain(29), country(34) # print(doc._.coref_chains.resolve(doc[31])) # Output: # # [Peter, wife]
View more
Author info

Richard Paul Hudson


Categories pipeline models standalone

Submit your project

If you have a project that you want the spaCy community to make use of, you can suggest it by submitting a pull request to the spaCy website repository. The Universe database is open-source and collected in a simple JSON file. For more details on the formats and available fields, see the documentation. Looking for inspiration your own spaCy plugin or extension? Check out the project idea label on the issue tracker.

Read the docsJSON source