MNIST Dataset

This example shows how to download/load/import MNIST

Out:

        ... mnist.pkl.gz already exists
Loading mnist
Dataset mnist loaded in 0.81s.

import symjax
import matplotlib.pyplot as plt

mnist = symjax.data.mnist()

plt.figure(figsize=(10, 4))
for i in range(10):
    plt.subplot(2, 5, 1 + i)
    plt.imshow(mnist["train_set/images"][i, 0], aspect="auto", cmap="Greys")
    plt.xticks([])
    plt.yticks([])
    plt.title(str(mnist["train_set/labels"][i]))

plt.tight_layout()