lale.util.numpy_to_torch_dataset module¶

class lale.util.numpy_to_torch_dataset.NumpyTorchDataset(X, y=None)[source]¶

Bases: torch.utils.data.Dataset

Pytorch Dataset subclass that takes a numpy array and an optional label array.