datasets

class pyraug.data.datasets.BaseDataset(digits, labels, binarize=False)

This class is the Base class for pyraug’s dataset

A __getitem__ is redefined and outputs a python dictionnary with the keys corresponding to data, labels etc… This Class should be used for any new data sets.

__getitem__(index)

Generates one sample of data

Parameters

index (int) – The index of the data in the Dataset

Returns

A dictionnary with the keys ‘data’ and ‘labels’ and corresponding torch.Tensor

Return type

(dict)