Text classification trainer for the BOW model.
- Python 3.5
- Tensorflow
- TFLearn
- 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"
Tensofrlow Simple Trainer is licensed under the MIT license.