栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

BERT 命名实体识别 (pytorch实现)

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

BERT 命名实体识别 (pytorch实现)

BERT命名实体识别代码链接:

GitHub - kamalkraj/BERT-NER: Pytorch-Named-Entity-Recognition-with-BERT

数据集train.text 中的数据格式:

 Dev数据集测试命令行:

python run_ner.py --data_dir=/data  --bert_model=bert-base-cased --task_name=ner --output_dir=out_basess --max_seq_length=128 --do_train --num_train_epochs 5 --do_eval --warmup_proportion=0.1

运行结果截图:

 

 

Test 数据集测试命令行: 

python run_ner.py --data_dir=/data  --bert_model=bert-base-cased --task_name=ner --output_dir=out_basestest --max_seq_length=128 --do_train --num_train_epochs 5 --do_eval --eval_on test --warmup_proportion=0.1 

运行结果截图:

  

 

从实验结果开看,我比链接中显示的结果要略差一点,F1值低一些。我这边服务器的GPU配置是两个1080ti。

注:

Accuracy(准确率):预测正确的元素个数/总的元素个数

Precision(查准率): 预测正确的实体个数/预测的实体总个数

Recall(召回率):预测正确的实体个数/标注的实体总个数

F1 值:F1 = 2 *准确率 * 召回率 / (准确率 + 召回率)

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/923219.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号