Create a new training config. :param device: the device to use ('cpu' or 'gpu') :param per_gpu_train_batch_size: the number

Create a new evaluation config. :param device: the device to use ('cpu' or 'gpu') :param n_gpu: the number of gpus to use

Create a new preprocessor. :param wrapper: the wrapper for the language model to use :param task_name: the name of the task

Create a new InputExample. :param guid: a unique textual identifier :param text_a: the sequence of text :param text_b

Create new InputFeatures. :param input_ids: the input ids corresponding to the original text or text sequence :param attentio

Implementation of a list that stores only a limited amount of examples per label. :param labels: the set of all possible labels

Create a new PVP. :param wrapper: the wrapper for the underlying language model :param pattern_id: the pattern id to use

Yield successive n-sized chunks from lst.

Compute the distillation loss (KL divergence between predictions and targets) as described in the PET paper

This method loads dev examples from a file with name `DEV_FILE_NAME` in the given directory. :param data_dir: the directory in which

Get a collection of `InputExample`s for the dev set.

This method returns all possible labels for the task.

Get the list of labels for this data set.

This function defines the actual patterns: It takes as input an example and outputs the result of applying a pattern to it. To allow

Get the inputs for sequence classification. Override this method if the input for the task considered is of a more complicated form t

This method loads test examples from a file with name `TEST_FILE_NAME` in the given directory. :param data_dir: the directory in whic

Get a collection of `InputExample`s for the test set.

This method loads train examples from a file with name `TRAIN_FILE_NAME` in the given directory. :param data_dir: the directory in wh

Get a collection of `InputExample`s for the train set.

This method loads unlabeled examples from a file with name `UNLABELED_FILE_NAME` in the given directory. :param data_dir: the directo

Get a collection of `InputExample`s for the unlabeled set.

Return `true` iff no more examples can be added to this list

Load a set of input examples from a file