xnuohz / TextCNN

Tensorflow implementation of "Convolutional Neural Networks for Sentence Classification"

TextCNN

Tensorflow implementation of "Convolutional Neural Networks for Sentence Classification"

Requirements

• python 3.6
• tensorflow 1.12.0
• tqdm 4.28.1
• click 7.0
• YAML

Usage

• preprocess

python -m cnn.preprocess --data-cnf ../config/dataset/rt-polarity.yaml

• training

python -m cnn.main --data-cnf ../config/dataset/rt-polarity.yaml --model-cnf ../config/model/CNN.yaml

Result

Reference

• Paper - Convolutional Neural Networks for Sentence Classification
• Tutorial - Implementing a CNN for Text Classification in TensorFlow