lale.util.batch_data_dictionary_dataset module¶

class lale.util.batch_data_dictionary_dataset.BatchDataDictDataset(X, y=None)[source]¶

Bases: torch.utils.data.Dataset

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