src.ml.dataset.StimulusDataset

class StimulusDataset(x_data, y_data, img_transform, class2idx)

Bases: torch.utils.data.torch.utils.data.Dataset._name

Torch dataset for stimulus images.

Parameters

• x_data – A 3-d numpy array of input stimulus image.

• y_data – A 1-d numpy array of output class labels. be “train” or “test”.

• img_transforms – A dictionary of torchvision transforms.

• class2idx – A mapping between class and it’s corresponding integer idx.

x_data

A 3-d numpy array of input stimulus image.

Type

np.ndarray

y_data

A 1-d numpy array of output class labels. be “train” or “test”.

Type

np.ndarray

img_transforms

A dictionary of torchvision transforms.

Type

dict

class2idx

A mapping between class and it’s corresponding integer idx.

Type

dict

Methods

__call__(*args, **kwargs)

Call self as a function.

Return type

Any