Querying Our Neural Network

Now that we have a fairly well-trained network, let’s ask it to classify images. We’ll switch to the MNIST test dataset of 10,000 images. These are images our neural network has not yet seen.

Loading the test data

Let’s load the dataset with a new Dataset object.