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

get_data()[source]
lale.util.pandas_to_torch_dataset.pandas_collate_fn(batch)[source]