I\'m trying to build a CNN model that to the best of my ability will classify correctly the MNIST dataset. I want to write a code that will select the best model and save it