example.py 文件源码

def load_mnist():
    url = "http://deeplearning.net/data/mnist/mnist.pkl.gz"
    mnist_compressed = "mnist.pkl.gz"

    if not exists(mnist_compressed):
        print("Downloading MNIST")
        urlretrieve(url, mnist_compressed)

    # Load the dataset
    with gzip.open(mnist_compressed, "rb") as f:
        u = pickle._Unpickler(f)
        u.encoding = "latin1"
        data = u.load()

    data = [(X.reshape(-1, 28, 28), y) for X, y in data]
    return data