Skip to content

本项目旨在通过Tensorflow基于BiLSTM+CRF实现中文分词、词性标注、命名实体识别(NER)。

Notifications You must be signed in to change notification settings

haif-liu/NLP-tools

Repository files navigation

NLP-tools

本项目旨在通过Tensorflow基于BiLSTM+CRF实现中文分词、词性标注、命名实体识别(NER)。

模型架构:输入层-->嵌入层-->双向长短记忆网络CELL-->输出层。

模型从网络中自我学习嵌入矩阵(非预训练)提高效率。

欢迎各位大佬吐槽。

模型训练

corpus文件下放入语料集,语料格式:人民网/nz 1月4日/t 讯/ng 据/p [法国/nsf 国际/n。

执行python train.py 开始训练。

生成word2id字典存入data/data.pkl。

训练生成checkpoint存入ckpt。

模型超参数

  • 分词标记方式:4tags

  • 嵌入层向量长度:100

  • BiLstm层数:2

  • 隐藏层节点数:256

  • 最大迭代次数:9

  • Batch宽度:128

  • 初始学习率:1e-4(采用动态形式,随训练进行而减小步长)

模型测试

将待分词项写入test/test.txt文件中,执行python model_test.py,生成结果存入test/test_result.txt。

现状

目前模型尚处于初步测试成功,分词部分完成,正确率94%(有待提高)。

后期陆续整理出POS\NER以及Parse功能。

参考

本项目模型BiLSTM+CRF参考论文:http://www.aclweb.org/anthology/N16-1030

About

本项目旨在通过Tensorflow基于BiLSTM+CRF实现中文分词、词性标注、命名实体识别(NER)。

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages