Skip to content

ProHiryu/bert-chinese-ner

Folders and files

NameName
Last commit message
Last commit date

Latest commit

96efd23 · Feb 26, 2020

History

17 Commits
Nov 28, 2018
Apr 11, 2019
Feb 26, 2020
Oct 16, 2019
Nov 28, 2018
Nov 28, 2018
Nov 28, 2018
Nov 28, 2018

Repository files navigation

bert-chinese-ner

前言

使用预训练语言模型BERT做中文NER尝试,fine - tune BERT模型

PS: 移步最新albert fine-tune ner模型

代码参考

使用方法

BERT-TF下载bert源代码,存放在路径下bert文件夹中

BERT-Base Chinese下载模型,存放在checkpoint文件夹下

使用BIO数据标注模式,使用人民日报经典数据

train:

python BERT_NER.py --data_dir=data/ --bert_config_file=checkpoint/bert_config.json --init_checkpoint=checkpoint/bert_model.ckpt --vocab_file=vocab.txt --output_dir=./output/result_dir/

结果

经过100个epoch跑出来的结果

eval_f = 0.9662649
eval_precision = 0.9668882
eval_recall = 0.9656949
global_step = 135181
loss = 40.160034

测试结果第一句:

About

使用预训练语言模型BERT做中文NER

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published