PeiPandarinathNLB2021

class torch_brain.datasets.PeiPandarinathNLB2021(root=None, recording_ids=None, transform=None, dirname='pei_pandarinath_nlb_2021', **kwargs)

Bases: torch_brain.datasets.mixins.SpikingDatasetMixin, torch_brain.datasets.dataset.Dataset

Curated spiking neural activity datasets from the Neural Latents Benchmark 2021 (NLB’21).

Preprocessing

To download and prepare this dataset, run

brainsets prepare pei_pandarinath_nlb_2021

get_sampling_intervals(split=None)

Returns a dictionary of sampling intervals for each recording. This represents the intervals that can be sampled from each session.

This dictionary will be used by torch_brain’s Samplers to know where to sample from.

The default method returns intervals containing the entire domain of each recording. This behavior can be overridden by subclasses to give out custom sampling intervals.

Returns:

Dictionary mapping recording IDs to their time domain intervals.