hellaswag.yaml 679 Bytes
Newer Older
nikuya3's avatar
nikuya3 committed
1
2
3
4
5
6
7
8
9
group:
  - multiple_choice
task: hellaswag
dataset_path: hellaswag
dataset_name: null
output_type: multiple_choice
training_split: train
validation_split: validation
test_split: null
10
doc_to_text: "{% set text = activity_label ~ ': ' ~ ctx_a ~ ' ' ~ ctx_b.capitalize() %}{{text|trim|replace(' [title]', '. ')|regex_replace('\\[.*?\\]', '')|replace('  ', ' ')}}"
lintangsutawika's avatar
lintangsutawika committed
11
12
doc_to_target: "{{label}}"
doc_to_choice: "{{endings|map('trim')|map('replace', ' [title]', '. ')|map('regex_replace', '\\[.*?\\]', '')|map('replace', '  ', ' ')|list}}"
nikuya3's avatar
nikuya3 committed
13
14
15
16
17
18
19
metric_list:
  - metric: acc
    aggregation: mean
    higher_is_better: true
  - metric: acc_norm
    aggregation: mean
    higher_is_better: true