/allennlp-finetune-ner

Fine Tune NER models with AllenNLP

Primary LanguagePython

Installation

conda create -n allennlp python=3.7
conda activate allennlp
pip install allennlp==1.3.0

Download English CoNLL

get_data.sh

Training

See scripts/sweep.sh

export LEARNING_RATE=2e-05
allennlp train configs/ner-bert.json -s experiments/exp-${LEARNING_RATE} --include-package mylib

Prediction

allennlp predict experiments/exp-2e-05/ conll2003/eng.testa --use-dataset-reader --cuda-device 0 --output-file out.txt