lm_eval/tasks/super_glue/wsc/t5-prompt.yaml (20 lines of code) (raw):
tag:
- super-glue-t5-prompt
task: super_glue-wsc-t5-prompt
dataset_path: super_glue
dataset_name: wsc.fixed
training_split: train
validation_split: validation
output_type: generate_until
doc_to_text: !function "t5_utils.doc_to_text"
process_results: !function "t5_utils.process_results"
doc_to_target: label
generation_kwargs:
until:
- "</s>"
metric_list:
- metric: accuracy
aggregation: mean
higher_is_better: true
metadata:
version: 1.0