gwexpy.interop.torch_dataset

Functions

to_torch_dataloader(dataset, *[, ...])	Create a torch DataLoader from the provided dataset.
to_torch_dataset(obj, *, window[, stride, ...])	Build a TimeSeriesWindowDataset from the provided object.

Classes

TimeSeriesWindowDataset(series, *, window[, ...])

Simple windowed dataset wrapper for torch training loops.

class gwexpy.interop.torch_dataset.TimeSeriesWindowDataset(series, *, window: int, stride: int = 1, horizon: int = 0, labels: TimeSeries | TimeSeriesMatrix | np.ndarray | Callable | None = None, multivariate: bool = False, align: str = 'intersection', device=None, dtype=None)

Bases: object

Simple windowed dataset wrapper for torch training loops.

gwexpy.interop.torch_dataset.to_torch_dataset(obj, *, window: int, stride: int = 1, horizon: int = 0, labels: TimeSeries | TimeSeriesMatrix | np.ndarray | Callable | None = None, multivariate: bool = False, align: str = 'intersection', device=None, dtype=None)

Build a TimeSeriesWindowDataset from the provided object.

gwexpy.interop.torch_dataset.to_torch_dataloader(dataset, *, batch_size: int = 1, shuffle: bool = False, num_workers: int = 0, **kwargs)

Create a torch DataLoader from the provided dataset.