RecconEmotionEntailmentData

class RecconEmotionEntailmentData(dataset: torch.utils.data.dataset.TensorDataset, is_training: bool = True)

Class to create torch Dataset instance, which is the required data type for Transformer’s Trainer

Parameters

• dataset (TensorDataset) – TensorDataset object

• is_training (bool, optional) – Set True if training, set False if evaluating. Defaults to True.