How to use BERT for ENTITY extraction from a Sequence without classification in the NER task?

kyzhouhzau / BERT-NER

Use Google's BERT for named entity recognition （CoNLL-2003 as the dataset）.

MIT License

1.24k stars 335 forks source link

How to use BERT for ENTITY extraction from a Sequence without classification in the NER task? #73

Closed ManojPrabhakar closed 4 years ago

ManojPrabhakar commented 4 years ago

My requirement here is given a sentence(sequence), I would like to just extract the entities present in the sequence without classifying them to a type in the NER task. I see that BertForTokenClassification for NER does the classification. Can this be adapted for just the extraction?

Can you give me an idea of how to do entity extraction/identification using BERT?