behavenet
latest
Contents:
Installation
User guide
Advanced user guide
BehaveNet data structure
Hyperparameter glossary
behavenet.data
behavenet.data.data_generator Module
Functions
Classes
SingleSessionDatasetBatchedLoad
SingleSessionDataset
ConcatSessionsGenerator
ConcatSessionsGeneratorMulti
behavenet.data.preprocess Module
behavenet.data.transforms Module
behavenet.data.utils Module
behavenet.fitting
behavenet.models
behavenet.plotting
behavenet
behavenet.data
SingleSessionDatasetBatchedLoad
Edit on GitHub
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.