src.ml.dataset.FMRIDataset

class FMRIDataset(x_data, y_data, class2idx)[source]

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

Torch dataset for fMRI data.

Parameters

  • x_data – A 2-d numpy array of input fmri data.

  • y_data – A 1-d numpy array of output class labels.

  • class2idx – A dictionary that maps class string to integers.

x_data

A 2-d numpy array of input fmri data.

y_data

A 1-d numpy array of output class labels.

class2idx

A dictionary that maps class string to integers.

Methods

__call__(*args, **kwargs)

Call self as a function.

Return type

Any