hookeai.time_series_data.time_dataset.get_parent_dataset_indices

get_parent_dataset_indices(subset, subset_indices)[source]

Get parent data set indices from subset indices.

Parameters:

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

  • subset_indices (list[int]) – Subset samples indices.

Returns:

parent_indices – Parent data set samples indices.

Return type:

list[int]