# TextClassify

**Repository Path**: mortal-wei/TextClassify

## Basic Information

- **Project Name**: TextClassify
- **Description**: 分别利用textcnn、textrnn、word2vec+lstm进行文本分类
- **Primary Language**: Python
- **License**: MIT
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 1
- **Forks**: 1
- **Created**: 2020-03-13
- **Last Updated**: 2022-08-14

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# 文本分类
## 数据集 
新闻分类数据集  
包含{'体育': 0, '娱乐': 1, '家居': 2, '房产': 3, '教育': 4, '时尚': 5, '时政': 6, '游戏': 7, '科技': 8, '财经': 9}十个类别，其中train文件50000条，val5000条，test10000条。  

## 算法实现
- bayes
- rnn
- cnn
- w2v+lstm
- attenton+cnn

## 代码说明
```
TextClassify/  
|-- data/                    # 数据文件
|   |-- raw_data             # 原始数据
|   |   |-- data_utils.py    # 文件预处理
|   |-- segment_data         # 经过处理后的数据（分词）
|-- model/                   # 存储训练结果文件（持久化模型、结构图、混淆矩阵、损失曲线）
|-- public/                  
|   |   |-- conf.py          # 分类模型配置文件
|   |   |-- plt.py           # 作图
|-- pretrain.model           # 词向量模型（根据需要去别处下载）
|-- Attention.py             # attention模块
|-- SimpleCNN.py             # 基础版cnn
|-- SimpleRNN.py             # 基础版rnn
|-- BI-LSTM.py               # bi-lstm + word2vec
|-- SklearnBayes.py          # 贝叶斯分类
|-- TextCNN.py               # textcnn
|-- AttentionCNN.py          # attention＋cnn
```