search

kyzhouhzau / BERT-NER

Use Google's BERT for named entity recognition （CoNLL-2003 as the dataset）.
MIT License
1.23k stars 335 forks source link

Extending to other NER labels? #43

Open alipetiwala opened 5 years ago

alipetiwala commented 5 years ago

Can we train using a custom NER tagset, currently in the code the IOB labels seem to be hardcoded?

kyzhouhzau commented 5 years ago

Of course, if you want to use your own tags. You should change #L227 https://github.com/kyzhouhzau/BERT-NER/blob/49e0f9f41ed7333d8fabe9e087e6006bd28fb20e/BERT_NER.py#L227