hookeai.time_series_data.time_dataset.change_dataset_features_labels

change_dataset_features_labels(dataset, features_label_map)

Change time series data set features labels.

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).

• features_label_map (dict) – Features labels mapping. For each original feature label (key, str), specifies the new label (item, str). Nonexistent features are silently ignored.

Returns:

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).

Return type:

torch.utils.data.Dataset