arc.yaml 627 Bytes
Newer Older
1
2
dataset_path: ai2_arc
dataset_name: ARC-Challenge
3
output_type: multiple_choice
4
5
6
training_split: train
validation_split: validation
test_split: test
7
8
template_aliases: "{% set answer_choices = choices['text'] %}{% set gold = choices.label.index(answerKey) %}" # set the list of possible answer choices, and set what this doc's gold answer is (set what ds column used, and what)
doc_to_text: "Question: {{question}}\nAnswer:"
haileyschoelkopf's avatar
haileyschoelkopf committed
9
doc_to_target: "{{gold}}" # this will be cast to an int. 
lintangsutawika's avatar
lintangsutawika committed
10
metric_list:
11
  - metric: acc
lintangsutawika's avatar
lintangsutawika committed
12
13
    aggregation: mean
    higher_is_better: true
14
15
16
  - metric: acc_norm
    aggregation: mean
    higher_is_better: true