中文命名实体识别迁移学习扩展版
1.首先把项目搭建好
参考这篇博文:
https://blog.csdn.net/weixin_43721000/article/details/119993664
2.然后进行扩展
参考这个项目:
https://gitee.com/smdg/named-entity-recognition-ner/blob/master/README.md
(1) 项目简单说明
把ner_run.py、service_run.py、templates、service_op.py放到项目跟目录
用albert_ner.py中的返回label数组 替换到项目文件albert_ner.py中的返回label数组
(2) 项目详细说明
迁移成为自用的ner需要做如下步骤:
1.加入这三个文件和一个文件夹
(1)ner_run.py是用来训练和预测的命令
(2)service_run.py 和 templates 文件夹是一个服务接口和前端页面文件,启动service_run.py,浏览器访问http://127.0.0.1:5000/machine_learning/ner/title_split?title=[“置身事内:中国政府”]
可以看到预测结果
(3)service_op.py 是输入和输出数据规范整理的一些方法
2.修改一个文件
将 albert_ner.py 文件的 get_labels()方法中的label修改成自己需要的label