/porn_fiction_classify

一个色情小说检测项目

Primary LanguageHTML

色情文章检测

简介

利用 文本卷积神经网络 (TextCNN)训练的文章分类模型,检测是否为色情文章。

Author:yudake
date:2018/2/22

项目详细解析可以参考我的这篇博文

考虑到社会主义核心价值观,没有上传训练集。不过data_processed中有处理过的数据集和分词集,可以直接使用。
或者可以自己寻找训练数据,根据自己的训练数据扩充分词集,然后利用本模型作迁移学习。

开发环境:

  • python3.x;
  • tensorflow1.2;
  • jieba;
  • 其他相关类库,程序内会有提及。

目标:实现色情文章检测。

模型效果:最终准确率在98%以上。

模型

系统模型

  • 输入:文章经过处理后的句子矩阵;
  • 模型:文本卷积神经网络;
  • 输出:分类结果。

准确率曲线

准确率

平均准确率在98%。具体内容请看代码与博客。