- Python 3.7
- virtualenv
- Create a virtual environment:
virtualenv venv
- Activate the virtual environment:
.\venv\Scripts\activate.ps1
(will vary depending on OS) - Install dependencies:
pip install -r requirements.txt
python classifier.py
classifier.py
will set up a Keras model and train it using the objective samples in data/objectives.csv
. Once trained, it outputs a few predictions, and then serializes the model and tokenizer for later use. (as objectives.h5
and tokenizer.pickle
)
python load_model.py
load_model.py
is an example of how to load a pre-trained model from file and make some predictions using it.