伯特·纳尔
使用google BERT进行CoNLL-2003 NER!
使用Python训练模型并使用C ++进行推理
要求
python3
pip3 install -r requirements.txt
跑步
python run_ner.py --data_dir=data/ --bert_model=bert-base-cased --task_name=ner --output_dir=out_base --max_seq_length=128 --do_train --num_train_epochs 5 --do_eval --warmup_proportion=0.1
结果
BERT库
验证数据
precision recall f1-score support
PER 0.9677 0.9745
1