Steps to run the code:
- Change the training directory at line 25 to train the data with the python script.
- Create folders named models and logs.
- Run
pip install -r requirements.txt
to get necessary packages. - Run
python model.py
to train the model. - Get the best model from models directory and predict!