src.ml.datasetΒΆ

Functions for dataset processing.

Functions

calculate_mean_std

Calculate dataset mean and standard deviation.

create_weighted_sampler

Create a weighted random sampler.

Classes

FMRIDataset

Torch dataset for fMRI data.

StimulusDataset

Torch dataset for stimulus images.