pydpf.datautils.StateSpaceSubset

class pydpf.datautils.StateSpaceSubset(datadict, parent: StateSpaceDataset)

Bases: StateSpaceDataset

__init__(datadict, parent: StateSpaceDataset)

Methods

__init__(datadict, parent)
apply(f[, modified_series])	Apply a function across all trajectories
collate(batch)	Pass to the collate_fn parameter of any torch.utils.data.DataLoader object that uses this dataset.
deterministic_split(ratios)
normalise_dims([normalised_series, ...])	Normalise the data to have mean zero and standard deviation one.
random_split(ratios, generator)
save(path[, series_metadata_path, ...])	Save a dataset to a file or folder
select(indices)

Attributes

control	The dataset control as a tensor
control_dimension	The dimension of the control actions
observation	The dataset observation as a tensor
observation_dimension	The dimension of each observation
series_metadata	The dataset series_metadata as a tensor
state	The dataset state as a tensor
state_dimension	The dimension of the state
time	The dataset time as a tensor