Skip to content

hiyouga/BiLSTM-CRF-PyTorch-demo

Repository files navigation

BiLSTM-CRF-PyTorch-demo

A simple baseline model for Named Entity Recognition.

Requirement

  • PyTorch 1.2.0
  • NumPy 1.17
  • Python 3.6
  • GloVe pre-trained word vectors:

Dataset

Based on the shared task of Named Entity Recognition on CoNLL 2003 (English corpus). [link]

Train

python train.py

Reference

  • Lample, G., Ballesteros, M., Subramanian, S., Kawakami, K., & Dyer, C. "Neural Architectures for Named Entity Recognition." Proceedings of NAACL-HLT (pp. 260-270). 2016. [pdf]

  • Linear-chain conditional random field implemented by kmkurn: pytorch-crf

License

MIT

Releases

No releases published

Packages

 
 
 

Languages