Run TensorFlow Models in the Browser

In this section we will train a simple digit recognition model using the MNIST dataset provided through the TensorFlow library². To load the dataset, in the first cell of you notebook type the following. This will display the shape of the training data inputs (tx) and target (ty), and validation data inputs (vx) and target (vy). Let's now display a set of 10 sample images for each digit to understand how the data looks like. This will generate the following grid of images.