.
├── README.md
├── models 模型文件
│ ├── Bert.py
│ ├── DPCNN.py
│ ├── FastText.py
│ ├── TextCNN.py
│ ├── TextRCNN.py
│ ├── TextRNN.py
│ ├── TextRNN_Att.py
│ └── Transformer.py
├── run.py
├── train_eval.py 训练代码
├── utils.py 工具代码
├── utils_fasttext.py
└── zhengzhuang 数据集
├── data
│ ├── class.txt
│ ├── dev.txt
│ ├── test.txt
│ ├── train.txt
│ └── vocab.pkl
├── log 日志
└── saved_dict 模型保存
数据集详细介绍见PTM,评测指标为Precision@N, Recall@N
环境,使用方法等详细介绍见网址Chinese-Text-Classification-Pytorch。
模型中有方法TextCNN, TextRNN, TextRNN_Att, TextRCNN, FastText, DPCNN, Transformer, Bert
,运行命令:
python run.py --model Bert