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