logitorch.datasets.te.control_dataset

Attributes

CONTROL_DATASET
CONTROL_DATASET_FOLDER
CONTROL_DATASET_ZIP_URL
CONTROL_ID_TO_LABEL
CONTROL_LABEL_TO_ID

Classes

ControlDataset

Helper class that provides a standard way to create an ABC using

Module Contents

class logitorch.datasets.te.control_dataset.ControlDataset(split_set: str)

Bases: logitorch.datasets.base.AbstractTEDataset

Helper class that provides a standard way to create an ABC using inheritance.

The function takes in a string as an argument and returns None.

The function first checks if the string is in the list of split sets. If it is not, it raises an error.

If the string is in the list of split sets, the function checks if the control dataset folder exists. If it does not, it downloads the dataset.

The function then sets the split set to the string that was passed in as an argument.

The function then sets the dataset path to the control dataset folder and the split set.

The function then reads the dataset and sets the premises, hypotheses, and labels to the read dataset.

If there is an error, the function prints the error message.

Parameters:

split_set (str) – str

logitorch.datasets.te.control_dataset.CONTROL_DATASET = 'control_dataset'

logitorch.datasets.te.control_dataset.CONTROL_DATASET_FOLDER = '/logitorch_datasets/control_dataset'

logitorch.datasets.te.control_dataset.CONTROL_DATASET_ZIP_URL = 'https://www.dropbox.com/s/rmcqituydxacuhv/control_dataset.zip?dl=1'

logitorch.datasets.te.control_dataset.CONTROL_ID_TO_LABEL

logitorch.datasets.te.control_dataset.CONTROL_LABEL_TO_ID