Skip to content

EdwardChan5000/tf-text-classification

 
 

Repository files navigation

CNN and RNN with Attention for Text Classification in Tensorflow

Requirements

  • Python 3
  • Tensorflow 1.0
  • Scikit-learn
  • Numpy
  • Pandas

Training models

CNN:

python train_cnn.py

RNN:

python train_rnn.py

Evaluating models

python eval.py --checkpoint_dir=./runs/textrnn/trained_result_1512462690/checkpoints --model_type=RNN

If evaluation data has labels, has_label should be set to True.

About

CNN and RNN with Attention for Text Classification in Tensorflow

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%