HMI/人机接口>> CHINESE_NER_PYTORCH>> 返回
项目作者： Fyfy1996

项目描述 ：
Pytorch Version for NER, Include LSTM (Random initialized Embeddings)+CRF or BERT Embeddings+CRF
高级语言： Python
项目地址: git://github.com/Fyfy1996/CHINESE_NER_PYTORCH.git
创建时间： 2021-03-22T07:10:27Z
项目社区：https://github.com/Fyfy1996/CHINESE_NER_PYTORCH
开源协议：
下载


CHINESE_NER_PYTORCH

Pytorch Version for NER, Name Entity Recognization

torch版本的命名实体识别

Tools including:

有如下的模型（脚本）

lstm + crf

bert + softmax

bert + crf

Requirements

主要使用的包，其他包（如numpy等）就不赘述

pytorch 1.8.0
transformers, 4.15.0

How to use

  1. 使用BIO标注
  2. 替换dataset文件夹的训练和测试txt
  3. 修改 run_MODEL_NAME.py 中的参数

  4. 使用如下命令运行模型

    1.  python run_lstmcrf.py

Results

Dataset文件夹中有一个样例数据，在此数据上，结果如下
||Precision|Recall|F1|
|———|:-:|:-:|:-:|
|LSTM+CRF|-|-|-|
|BERT+SOFTMAX|91.65|92.13|91.18|
|BERT+CRF|91.60|91.95|91.77|

其他

Feel Free to use it！
如有问题和bug等，欢迎交流 yonttavan@126.com