Text Classification

cnn、rnn、rcnn、rcnn_attention、transformer based pre-trained word embedding
bert，albert，ernie based char and pre-trained language model

Description

Download pre-trained language model and chinese news dataTHUCNews

How to use

# classification based  pre-trained word embedding
python run.py --model cnn --data_conf conf/data.yaml

# bert or else based char
python run.py --model bert --data_conf conf/data.yaml

Preferences

[1] BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding
[2] Convolutional Neural Networks for Sentence Classification
[3] Recurrent Convolutional Neural Networks for Text Classification [4] attention-is-all-you-need-pytorch