Refactoring: split into two scripts. One to just load the usps dataset and another one to create, train and test the neural network