SingleSessionDatasetBatchedLoad

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

Bases: Dataset

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