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