Text Classification Models with Tensorflow

Tensorflow implementation of Text Classification Models.

Implemented Models: Very Deep CNN [paper] Implementation of Very Deep Convolutional Networks for Text Classification.

Requirements

• Python3
• Tensorflow
• pip install -r requirements.txt

Usage

Merger data from repo craw_and_text_classification

run 1_prepare_text.py to make text file

python 1_prepare_text.py ../data_5/ ../data_merger/

you need specify alphabet by yourself, or use my define alphabet for Vietnamese in data_utils.py

python 2_get_alphabet.py ../data_merger/train.txt ../data_merger/alphabet1.txt
python 2_get_alphabet.py ../data_merger/test.txt ../data_merger/alphabet2.txt

and merger by using 3_merger_alphabet.py or handmade
To use my custom data, you can get it from data_merger

Train

python train.py ../data_merger/train.txt

Test

To test classification accuracy for test data after training,

python test.py ../data_merger/test.txt

References

• dennybritz/cnn-text-classification-tf
• zonetrooper32/VDCNN