
基于BERT的PyTorch命名实体识别(NER)源码
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
这段简介可以描述为:“基于BERT的PyTorch命名实体识别(NER)源码”是一个利用预训练语言模型BERT和深度学习框架PyTorch实现的高效准确的自然语言处理工具,专用于从文本中自动抽取实体信息。
伯特·纳尔使用Google的BERT进行CoNLL-2003命名实体识别任务。通过Python训练模型,并用C++进行推理。安装所需库需执行如下命令:`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库在验证数据上的表现如下:PER类别的精确度为0.9677,召回率为0.9745。
全部评论 (0)
还没有任何评论哟~


