RecconSpanExtractionData

class RecconSpanExtractionData(dataset: torch.utils.data.dataset.TensorDataset, for_predict: bool = False)

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

Parameters

• dataset (TensorDataset) – TensorDataset object

• for_predict (bool, optional) – Option to set for predict. Defaults to False.