MNIST Labels

class multivae.data.datasets.MnistLabels(data_path, split='train', download=False, random_binarized=True, dtype=torch.float32)

Mnist-Labels dataset. The first modality is the image and the second modality is the label.

transform_for_plotting(tensor, modality)

Transforms the label modality to text for plotting.