Datasets Utilities

Dataset loading utilities, mainly relying on the tensorflow_datasets module.

Datasets:

  • MNIST

datasets_utils.load_mnist(batch_size=50)

Load the MNIST train and test dataset

Parameters

batch_size (int, optional) – minibatch size, defaults to 50

Returns

mnist dataset

Return type

tf.Dataset