YCG09 / tf-text-classification

CNN and RNN with Attention for Chinese Text Classification in Tensorflow

Environment setup

• Requirements: python3, tensorflow1.10, sklearn, numpy, pandas and jieba

pip install -r requirements.txt

Training

• CNN

python train_cnn.py

• RNN with attention

python train_rnn.py

the data format is as follows: label '\t' text

Testing

• evaluation mode:

python test.py --checkpoint_dir=./runs/textcnn/trained_result_1548144557/checkpoints --model_type=CNN

• prediction mode:

python test.py --checkpoint_dir=./runs/textcnn/trained_result_1548144557/checkpoints --model_type=CNN --test_mode=prediction

Inference

python inference.py --checkpoint_dir=./runs/textrnn/trained_result_1548145204/checkpoints --model_type=RNN