PyTorch Implementation of NER with pretrained Bert

I know that you know BERT. In the great paper, the authors claim that the pretrained models do great in NER. It's even impressive, allowing for the fact that they don't use any prediction-conditioned algorithms like CRFs. We try to reproduce the result in a simple manner.

Requirements

• python>=3.6 (Let's move on to python 3 if you still use python 2)
• pytorch==1.0
• pytorch_pretrained_bert==0.6.1
• numpy>=1.15.4

Training & Evaluating

• STEP 1. Run the command below to download conll 2003 NER dataset.

bash download.sh

It should be extracted to conll2003/ folder automatically.

• STEP 2a. Run the command if you want to do the feature-based approach.

python train.py --logdir checkpoints/feature --batch_size 128 --top_rnns --lr 1e-4 --n_epochs 30

• STEP 2b. Run the command if you want to do the fine-tuning approach.

python train.py --logdir checkpoints/finetuning --finetuning --batch_size 32 --lr 5e-5 --n_epochs 3

Results in the paper

• Feature-based approach

• Fine-tuning

Results

• F1 scores on conll2003 valid dataset are reported.
• You can check the classification outputs in checkpoints.