SingleSessionDatasetBatchedLoad

class behavenet.data.data_generator.SingleSessionDatasetBatchedLoad(data_dir, lab='', expt='', animal='', session='', signals=None, transforms=None, paths=None, device='cpu', as_numpy=False)

Bases: torch.utils.data.dataset.Dataset

Dataset class for a single session with batch loading of data.