A simple baseline model for Named Entity Recognition.
- PyTorch 1.2.0
- NumPy 1.17
- Python 3.6
- GloVe pre-trained word vectors:
- Download pre-trained word vectors here.
- Extract the glove.840B.300d.zip to the
/glove/
folder.
Based on the shared task of Named Entity Recognition on CoNLL 2003 (English corpus). [link]
python train.py
-
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
MIT