hookeai.miscellaneous.dataset_processing.convert_to_tsdataset.save_dataset

save_dataset(dataset, dataset_basename, dataset_directory, is_append_n_sample=True)[source]

Save PyTorch time series data set to file.

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

  • dataset_basename (str) – Data set file basename.

  • dataset_directory (str) – Directory where the times series data set is stored.

  • is_append_n_sample (bool, default=True) – If True, then data set size (number of samples) is appended to data set filename.

Returns:

dataset_file_path – PyTorch data set file path.

Return type:

str