`torchfilter.data._subsequence_dataset`

Private module; avoid importing from directly.

Module Contents

A data preprocessor for producing training subsequences from

class torchfilter.data._subsequence_dataset.SubsequenceDataset(trajectories: List[types.TrajectoryNumpy], subsequence_length: int)[source]

Bases: torch.utils.data.Dataset

A data preprocessor for producing training subsequences from a list of trajectories.

Thin wrapper around torchfilter.data.split_trajectories().

Parameters:

trajectories (list) – list of trajectories, where each is a tuple of (states, observations, controls). Each tuple member should be either a numpy array or dict of numpy arrays with shape (T, ...).
subsequence_length (int) – # of timesteps per subsequence.

__getitem__(self, index: int) → types.TrajectoryNumpy[source]

Get a subsequence from our dataset.

Parameters:: index (int) – Subsequence number in our dataset.
Returns:: tuple – (states, observations, controls) tuple that contains data for a single subsequence. Each tuple member should be either a numpy array or dict of numpy arrays with shape (subsequence_length, ...).

__len__(self) → int[source]

Total number of subsequences in the dataset.