├── .gitignore ├── ERNIE_pretrain └── README.md ├── README.md ├── THUCNews ├── data │ ├── class.txt │ ├── dev.txt │ ├── embedding_SougouNews.npz │ ├── embedding_Tencent.npz │ ├── test.txt │ ├── train.txt │ └── vocab.pkl └── saved_dict │ └── README.md ├── bert_pretrain └── README.md ├── models ├── DPCNN.py ├── ERNIE.py ├── FastText.py ├── TextCNN.py ├── TextRCNN.py ├── TextRNN.py ├── TextRNN_Att.py ├── Transformer.py ├── bert.py ├── bert_CNN.py ├── bert_DPCNN.py ├── bert_RCNN.py └── bert_RNN.py ├── predict.py ├── pretrain_eval.py ├── pretrain_predict.py ├── pretrain_run.py ├── pretrain_utils.py ├── pytorch_pretrained ├── __init__.py ├── __main__.py ├── convert_gpt2_checkpoint_to_pytorch.py ├── convert_openai_checkpoint_to_pytorch.py ├── convert_tf_checkpoint_to_pytorch.py ├── convert_transfo_xl_checkpoint_to_pytorch.py ├── file_utils.py ├── modeling.py ├── modeling_gpt2.py ├── modeling_openai.py ├── modeling_transfo_xl.py ├── modeling_transfo_xl_utilities.py ├── optimization.py ├── optimization_openai.py ├── tokenization.py ├── tokenization_gpt2.py ├── tokenization_openai.py └── tokenization_transfo_xl.py ├── run.py ├── train_eval.py ├── utils.py └── utils_fasttext.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/.gitignore -------------------------------------------------------------------------------- /ERNIE_pretrain/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/ERNIE_pretrain/README.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/README.md -------------------------------------------------------------------------------- /THUCNews/data/class.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/THUCNews/data/class.txt -------------------------------------------------------------------------------- /THUCNews/data/dev.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/THUCNews/data/dev.txt -------------------------------------------------------------------------------- /THUCNews/data/embedding_SougouNews.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/THUCNews/data/embedding_SougouNews.npz -------------------------------------------------------------------------------- /THUCNews/data/embedding_Tencent.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/THUCNews/data/embedding_Tencent.npz -------------------------------------------------------------------------------- /THUCNews/data/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/THUCNews/data/test.txt -------------------------------------------------------------------------------- /THUCNews/data/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/THUCNews/data/train.txt -------------------------------------------------------------------------------- /THUCNews/data/vocab.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/THUCNews/data/vocab.pkl -------------------------------------------------------------------------------- /THUCNews/saved_dict/README.md: -------------------------------------------------------------------------------- 1 | 该文件夹存放训练的模型 -------------------------------------------------------------------------------- /bert_pretrain/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/bert_pretrain/README.md -------------------------------------------------------------------------------- /models/DPCNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/models/DPCNN.py -------------------------------------------------------------------------------- /models/ERNIE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/models/ERNIE.py -------------------------------------------------------------------------------- /models/FastText.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/models/FastText.py -------------------------------------------------------------------------------- /models/TextCNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/models/TextCNN.py -------------------------------------------------------------------------------- /models/TextRCNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/models/TextRCNN.py -------------------------------------------------------------------------------- /models/TextRNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/models/TextRNN.py -------------------------------------------------------------------------------- /models/TextRNN_Att.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/models/TextRNN_Att.py -------------------------------------------------------------------------------- /models/Transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/models/Transformer.py -------------------------------------------------------------------------------- /models/bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/models/bert.py -------------------------------------------------------------------------------- /models/bert_CNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/models/bert_CNN.py -------------------------------------------------------------------------------- /models/bert_DPCNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/models/bert_DPCNN.py -------------------------------------------------------------------------------- /models/bert_RCNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/models/bert_RCNN.py -------------------------------------------------------------------------------- /models/bert_RNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/models/bert_RNN.py -------------------------------------------------------------------------------- /predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/predict.py -------------------------------------------------------------------------------- /pretrain_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/pretrain_eval.py -------------------------------------------------------------------------------- /pretrain_predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/pretrain_predict.py -------------------------------------------------------------------------------- /pretrain_run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/pretrain_run.py -------------------------------------------------------------------------------- /pretrain_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/pretrain_utils.py -------------------------------------------------------------------------------- /pytorch_pretrained/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/pytorch_pretrained/__init__.py -------------------------------------------------------------------------------- /pytorch_pretrained/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/pytorch_pretrained/__main__.py -------------------------------------------------------------------------------- /pytorch_pretrained/convert_gpt2_checkpoint_to_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/pytorch_pretrained/convert_gpt2_checkpoint_to_pytorch.py -------------------------------------------------------------------------------- /pytorch_pretrained/convert_openai_checkpoint_to_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/pytorch_pretrained/convert_openai_checkpoint_to_pytorch.py -------------------------------------------------------------------------------- /pytorch_pretrained/convert_tf_checkpoint_to_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/pytorch_pretrained/convert_tf_checkpoint_to_pytorch.py -------------------------------------------------------------------------------- /pytorch_pretrained/convert_transfo_xl_checkpoint_to_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/pytorch_pretrained/convert_transfo_xl_checkpoint_to_pytorch.py -------------------------------------------------------------------------------- /pytorch_pretrained/file_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/pytorch_pretrained/file_utils.py -------------------------------------------------------------------------------- /pytorch_pretrained/modeling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/pytorch_pretrained/modeling.py -------------------------------------------------------------------------------- /pytorch_pretrained/modeling_gpt2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/pytorch_pretrained/modeling_gpt2.py -------------------------------------------------------------------------------- /pytorch_pretrained/modeling_openai.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/pytorch_pretrained/modeling_openai.py -------------------------------------------------------------------------------- /pytorch_pretrained/modeling_transfo_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/pytorch_pretrained/modeling_transfo_xl.py -------------------------------------------------------------------------------- /pytorch_pretrained/modeling_transfo_xl_utilities.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/pytorch_pretrained/modeling_transfo_xl_utilities.py -------------------------------------------------------------------------------- /pytorch_pretrained/optimization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/pytorch_pretrained/optimization.py -------------------------------------------------------------------------------- /pytorch_pretrained/optimization_openai.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/pytorch_pretrained/optimization_openai.py -------------------------------------------------------------------------------- /pytorch_pretrained/tokenization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/pytorch_pretrained/tokenization.py -------------------------------------------------------------------------------- /pytorch_pretrained/tokenization_gpt2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/pytorch_pretrained/tokenization_gpt2.py -------------------------------------------------------------------------------- /pytorch_pretrained/tokenization_openai.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/pytorch_pretrained/tokenization_openai.py -------------------------------------------------------------------------------- /pytorch_pretrained/tokenization_transfo_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/pytorch_pretrained/tokenization_transfo_xl.py -------------------------------------------------------------------------------- /run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/run.py -------------------------------------------------------------------------------- /train_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/train_eval.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/utils.py -------------------------------------------------------------------------------- /utils_fasttext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JackHCC/Chinese-Text-Classification-PyTorch/HEAD/utils_fasttext.py --------------------------------------------------------------------------------