hookeai.miscellaneous.pruning.pruning_dataset.load_full_dataset

load_full_dataset(full_dataset_dir)

Load full data set.

Parameters:

full_dataset_dir (str) – Full data set directory.

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