lm_eval/tasks/super_glue/wic/t5-prompt.yaml (22 lines of code) (raw):

tag: - super-glue-t5-prompt task: super_glue-wic-t5-prompt dataset_path: super_glue dataset_name: wic training_split: train validation_split: validation output_type: generate_until doc_to_text: "wic sentence1: {{sentence1}} sentence2: {{sentence2}} word: {{word}}" doc_to_target: label doc_to_choice: ['False', 'True'] generation_kwargs: until: - "</s>" metric_list: - metric: exact_match aggregation: mean higher_is_better: true ignore_case: true ignore_punctuation: true metadata: version: 0.0