dataset_preparer/resolve_coref.py

14 lines
361 B
Python
Executable File

#!/usr/bin/env python
import neuralcoref
import spacy
import sys
def resolve_coref(doc_text, doc_label=""):
'''removes coreferences in doc_text using neuralcoref'''
nlp = spacy.load('en_core_web_sm')
neuralcoref.add_to_pipe(nlp)
return nlp(doc_text)._.coref_resolved
if __name__ == "__main__":
print(resolve_coref(sys.stdin.read()))