lale.util.batch_data_dictionary_dataset module

class lale.util.batch_data_dictionary_dataset.BatchDataDict(*args: Any, **kwargs: Any)[source]

Bases: Dataset

Pytorch Dataset subclass that takes a dictionary of format {‘<batch_idx>’: <batch_data>}.

X is the dictionary dataset and y is ignored.

Parameters

  • X (dict) – Dictionary of format {‘<batch_idx>’: <batch_data>}

  • y (None) – Ignored.