利用 文本卷积神经网络 (TextCNN)训练的文章分类模型,检测是否为色情文章。
Author:yudake
date:2018/2/22
项目详细解析可以参考我的这篇博文
考虑到社会主义核心价值观,没有上传训练集。不过data_processed中有处理过的数据集和分词集,可以直接使用。
或者可以自己寻找训练数据,根据自己的训练数据扩充分词集,然后利用本模型作迁移学习。
开发环境:
- python3.x;
- tensorflow1.2;
- jieba;
- 其他相关类库,程序内会有提及。
目标:实现色情文章检测。
模型效果:最终准确率在98%以上。
- 输入:文章经过处理后的句子矩阵;
- 模型:文本卷积神经网络;
- 输出:分类结果。
平均准确率在98%。具体内容请看代码与博客。