laion_fmri.torch_data

PyTorch Dataset integration for laion_fmri.

Classes

LaionFMRIDataset(subject, session[, roi, ...])

PyTorch Dataset wrapping one session of a LAION-fMRI subject.
class laion_fmri.torch_data.LaionFMRIDataset(subject, session, roi=None, mask=None, nc_threshold=None, image_transform=None)[source]

Bases: object

PyTorch Dataset wrapping one session of a LAION-fMRI subject.

Parameters:

  • subject (Subject) – A loaded Subject instance.

  • session (str) – BIDS session ID. Required – single-trial betas are stored per session.

  • roi (str or None) – ROI name for voxel selection.

  • mask (np.ndarray[bool] or None) – Custom boolean mask.

  • nc_threshold (float or None) – Minimum noise ceiling for voxel inclusion.

  • image_transform (callable or None) – Transform applied to image tensors.