search

thunlp / CorefBERT

Source code for EMNLP 2020 paper "Coreferential Reasoning Learning for Language Representation"
MIT License
67 stars 6 forks source link

Test coreference resolution on raw text. #6

Closed aakashb95 closed 3 years ago

aakashb95 commented 3 years ago

I have downloaded the base pretrained model from google drive and want to test it on the following raw text: "My dad and my mum are working hard. She is always nice but he is sometimes rude."

How should i go about using the model to predict coreferences on above text?

I would then want to resolve the coreferences. I have used this function from allenai for SpanBERT model.

YeDeming commented 3 years ago

Sorry, CorefBERT cannot do end-to-end coreference resolution. It can solve the coreference of GAP format. I recommand you to use this code https://github.com/mandarjoshi90/coref for end-to-end coreference resolution

aakashb95 commented 3 years ago

Thanks for the prompt response.