arc_easy_generation.yaml 1.14 KB
Newer Older
Baber Abbasi's avatar
Baber Abbasi committed
1
tag:
2
3
4
  - llama3_tasks
  - ai2_arc_generation
task: arc_east_generation
Baber Abbasi's avatar
Baber Abbasi committed
5
dataset_path: allenai/ai2_arc
6
dataset_name: ARC-Easy
Baber Abbasi's avatar
Baber Abbasi committed
7
8
9
10
11
12
output_type: generate_until
training_split: train
validation_split: validation
test_split: test
fewshot_split: train
doc_to_text: 'Given the following question and four candidate answers (A, B, C and D), choose the best answer.\nQuestion: {{question.strip()}}\nA. {{choices.text[0]}}\nB. {{choices.text[1]}}\nC. {{choices.text[2]}}{% if choices.text|length > 3 %}\nD. {{choices.text[3]}}{% endif %}\nYour response should end with "The best answer is [the_answer_letter]" where the [the_answer_letter] is one of A, B, C or D.'
Baber Abbasi's avatar
Baber Abbasi committed
13
gen_prefix: 'The best answer is'
Baber Abbasi's avatar
Baber Abbasi committed
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
fewshot_delimiter: "\n\n"
doc_to_target: "{{ 'ABCD'[answerKey|int - 1] if answerKey|string in '1234' else answerKey }}"
num_fewshot: 0
generation_kwargs:
  max_gen_toks: 100
  until:
    - "\n\n"
    - "."
metric_list:
  - metric: exact_match
    aggregation: mean
    higher_is_better: true
    ignore_case: true
    ignore_punctuation: true
filter_list:
  - name: remove_whitespace
    filter:
      - function: remove_whitespace
      - function: take_first
metadata:
  version: 1.0