Repository Layout The best model is in best_model, note that it is managed by git LFS so you may need to set that up. Data loading code is in my_datasets.py Training code is in training.py Final predicting code is in final.py The other files are utilities.