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.