Tensorflow Simple Trainer

Text classification trainer for the BOW model.

Requirements

• Python 3.5
• Tensorflow
• TFLearn

Usage

• Download the repository
• Install the requirements: pip install -r requirements.txt
• Enter your classes into classes.yml
• Enter some test data into test-data.yml
• Run python train.py to train your model
• Run python test.py to test your model
• Run python evaluate.py "It was a really bad movie"

License

Tensofrlow Simple Trainer is licensed under the MIT license.