Skip to main content

vame.model.dataloader

Variational Animal Motion Embedding 0.1 Toolbox © K. Luxem & P. Bauer, Department of Cellular Neuroscience Leibniz Institute for Neurobiology, Magdeburg, Germany

https://github.com/LINCellularNeuroscience/VAME Licensed under GNU General Public License v3.0

SEQUENCE_DATASET Objects​

class SEQUENCE_DATASET(Dataset)

__init__​

def __init__(path_to_file: str, data: str, train: bool, temporal_window: int,
             **kwargs) -> None

Initialize the Sequence Dataset.

Arguments:

  • path_to_file str - Path to the dataset files.
  • data str - Name of the data file.
  • train bool - Flag indicating whether it's training data.
  • temporal_window int - Size of the temporal window.

Returns:

None

__len__​

def __len__() -> int

Return the number of data points.

Returns:

  • int - Number of data points.

__getitem__​

def __getitem__(index: int) -> torch.Tensor

Get a normalized sequence at the specified index.

Arguments:

  • index int - Index of the item.

Returns:

  • torch.Tensor - Normalized sequence data at the specified index.