tsgm.dataset
¶
Module Contents¶
- class DatasetProperties(N: int, D: int, T: int, variables: DatasetProperties.__init__.T[DatasetProperties.__init__.T] = None)[source]¶
Stores the properties of a dataset. Along with dimensions it can store properties of the covariates.
- class Dataset(x: tsgm.types.Tensor, y: tsgm.types.Tensor, metadata: Dict | None = None)[source]¶
Bases:
DatasetProperties
Wrapper for time-series datasets. Additional information is stored in
Distribution metadata
field.- Parameters:
x (tsgm.types.Tensor) – The matrix of time series with dimensions NxDxT
y (tsgm.types.Tensor) – The lables of a time series.
metadata – Additional info for the dataset.
- property X: tsgm.types.Tensor[source]¶
Returns the time series tensor in format: n_samples x seq_len x feat_dim.