Skip to content

EdwardChan5000/tf-text-classification

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

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

 
 
 

Contributors

Languages

  • Python 100.0%