Dhanachandra / bert_crf

BERT CRF model for Name Entity Recognition in pytorch
MIT License
28 stars 15 forks source link
bert-crf-model entity-recognition pytorch

BERT_CRF

BERT_CRF model for Name Entity Recognition in pytorch
Dataset source: https://github.com/synalp/NER
conlleval.py downloaded from: https://github.com/sighsmile/conlleval

For traninig a model: python bert_crf.py --mode train
For testing the model: python bert_crf.py --mode test
You can check the notebook at https://github.com/Dhanachandra/bert-crf4NER/blob/master/bert-crf4NER/bert_crf_4_ner_training.ipynb