plugins.hf_seq_classification.data_classes

HfSeqClassificationDataset Objects

class HfSeqClassificationDataset(Dataset)

PyTorch Dataset.

__init__

def __init__(args, input_filepath, label_map)

Arguments:

• args - DataInterface arguments
• input_filepath str - Path to dataset
• label_map dict - Map categorical values to numerical

HfSeqClassificationDataInterface Objects

class HfSeqClassificationDataInterface(data_interface.DataInterface)

Retrieves train and val PyTorch Datasets.

__init__

def __init__(args)

Arguments:

• args arguments.DataArguments - Dataclass