torch_timeseries.dataloader.MaskTS

class MaskTS(dataset: TimeSeriesDataset, scaler: Scaler, time_enc=0, window: int = 168, mask_rate=0.4, scale_in_train=True, shuffle_train=True, freq=None, batch_size: int = 32, train_ratio: float = 0.7, val_ratio: float = 0.2, num_worker: int = 3, uniform_eval=True)[source]

Bases: object

Data loader for imputation time series datasets.

dataset

Time series dataset to be used.

Type:: TimeSeriesDataset

scaler

Scaler to normalize the data.

Type:: Scaler

time_enc

Time encoding flag.

Type:: int

window

Window size for the time series data.

Type:: int

mask_rate

Rate at which data is masked.

Type:: float

scale_in_train

Whether to scale data during training.

Type:: bool

shuffle_train

Whether to shuffle the training data.

Type:: bool

freq

Frequency of the time series data.

Type:: str or None

batch_size

Number of samples per batch.

Type:: int

train_ratio

Ratio of the dataset to be used for training.

Type:: float

val_ratio

Ratio of the dataset to be used for validation.

Type:: float

num_worker

Number of worker threads for data loading.

Type:: int

uniform_eval

Whether to use uniform evaluation.

Type:: bool

train_loader

DataLoader for the training data.

Type:: DataLoader

val_loader

DataLoader for the validation data.

Type:: DataLoader

test_loader

DataLoader for the test data.

Type:: DataLoader