Skip to content

Text classification using different neural networks.

Notifications You must be signed in to change notification settings

cz-rl/TextClassification

 
 

Repository files navigation

TextClassification

Text classification using different neural networks. 中文文本分类,使用TensorFlow 2.x实现TextCNN,TextRNN,TextRCNN,BiLSTM Attention, HAN等类型的深度学习模型, 。

数据

数据来源于 搜狗新闻数据
我们取新闻中的五个类别:汽车、娱乐、军事、体育、科技 将五个类别分词后保存到data文件目录,作为分类模型构建与验证数据

数据集 数据量
总数据 87747
训练集 65810
测试集 21973

环境

python 3.7 TensorFlow 2.0+

使用说明

进入到相关模型目录下

# 使用默认参数运行
python main.py

参数

进入到相关模型目录下

# 查看模型相关参数
python main.py -h

About

Text classification using different neural networks.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%