hookeai.model_architectures.rnn_base_model.train.training.get_time_series_data_loader

get_time_series_data_loader(dataset, batch_size=1, is_shuffle=False, **kwargs)[source]

Get time series data set data loader.

Parameters:

  • dataset (torch.utils.data.Dataset) – Time series data set. Each sample is stored as a dictionary where each feature (key, str) data is a torch.Tensor(2d) of shape (sequence_length, n_features).

  • batch_size (int, default=1) – Number of samples loaded per batch.

  • is_shuffle (bool, default=False) – Reshuffle data set at every epoch.

  • **kwargs – Arguments of torch.utils.data.DataLoader.

Returns:

data_loader – Time series data set data loader.

Return type:

torch.utils.data.DataLoader