pe.data.image.mnist module

class pe.data.image.mnist.MNIST(split='train')

Bases: Data

The MNIST dataset.

__init__(split='train')

Constructor.

Parameters:

split (str, optional) – The split of the dataset. It should be either “train” or “test”, defaults to “train”

Raises:

ValueError – If the split is invalid