wdataloader

Classes

USPS06Dataset

Dataset class for filtered USPS data (digits 0-6) stored in H5 file.

Module Contents

class wdataloader.USPS06Dataset(h5_file='./data/raw/usps.h5', set_type='train', transform=None)

Bases: torch.utils.data.Dataset

Dataset class for filtered USPS data (digits 0-6) stored in H5 file.

h5_file = './data/raw/usps.h5'

set_type = 'train'

transform = None

__len__()

get_input_dim(): Returns the input dimension of the dataset samples.

__getitem__(idx)