canonical_sets.data.base.DataSet

class DataSet(data, labels)[source]

Bases: Dataset

The dataset class for the PyTorch dataloader.

Initialize the dataset. :type data: DataFrame :param data: The data. :type data: pd.DataFrame :type labels: DataFrame :param labels: The labels. :type labels: pd.DataFrame

Methods