
PyTorch LSTM-CRF: 实现命名实体识别的代码库
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本项目提供了一个基于PyTorch框架实现的LSTM-CRF模型,用于执行高效的命名实体识别任务。代码简洁易懂,适合自然语言处理研究者和开发人员参考学习。
该存储库实现了用于命名实体识别的LSTM-CRF模型。此模型与另一模型相似,只是我们省略了BiLSTM之后的最后一个tanh层。我们在CoNLL-2003和OntoNotes 5.0英文数据集上均达到了最先进的性能(请通过使用Glove和ELMo来检查我们的结果,并通过对BERT进行微调以查看其他人的结果)。此外,我们实现了允许O(log N)推断和回溯的模块。
以下是模型在不同情况下的表现:
- 基于BERT的情况 + CRF:在CONLL-2003数据集上的精确度为91.69%,召回率为92.05%,F1分数为91.87%;在OntoNotes 5.0数据集上,精确度为89.57%,召回率89.45%
- Roberta-base + CRF:在CONLL-2003数据集上的精确度为91.88%,召回率为93.01%,F1分数为92.44%
全部评论 (0)
还没有任何评论哟~


