lale.util.pandas_to_torch_dataset module¶
- class lale.util.pandas_to_torch_dataset.PandasTorchDataset(*args: Any, **kwargs: Any)[source]¶
Bases:
Dataset
Pytorch Dataset subclass that takes a pandas DataFrame and an optional label pandas Series.
X and y are the dataset and labels respectively.
- Parameters
X (pandas DataFrame) – Two dimensional dataset of input features.
y (pandas Series) – Labels