ttslearn.train_util.get_data_loaders

ttslearn.train_util.get_data_loaders(data_config, collate_fn)[source]

Get data loaders for training and validation.

Parameters
  • data_config (dict) – Data configuration.

  • collate_fn (callable) – Collate function.

Returns

Data loaders.

Return type

dict