Here’s a copy of the final code we used to train and test our model (attached as a Jupyter notebook).
Download Jupyter Notebook