Function get_output_func

tasks/superglue/dataset.py:48–57

(task_name, args)

def get_output_func(task_name, args):
    def default_output_func(predictions, examples, output_file):
        with open(output_file, "w") as output:
            for prediction, example in zip(predictions, examples):
                data = {"idx": example["idx"], "label": prediction}
                output.write(json.dumps(data) + "\n")
    if task_name in PROCESSORS:
        return PROCESSORS[task_name](args).output_prediction
    else:
        return default_output_func


def read_tsv(path, **kwargs):