gsm8k.yaml 806 Bytes
Newer Older
1
2
group:
  - math_word_problems
3
task: gsm8k
4
5
dataset_path: gsm8k
dataset_name: main
6
output_type: generate_until
7
8
9
10
11
12
13
14
15
16
training_split: train
fewshot_split: train
test_split: test
doc_to_text: "Question: {{question}}\nAnswer:"
doc_to_target: "{{answer}}" #" {{answer.split('### ')[-1].rstrip()}}"
metric_list:
  - metric: exact_match
    aggregation: mean
    higher_is_better: true
    ignore_case: true
17
    ignore_whitespace: true
18
19
20
21
22
23
24
25
26
27
28
    ignore_punctuation: false
    regexes_to_ignore:
      - ","
      - "\\$"
      - ".*### "
generation_kwargs:
  until:
    - "\n\n"
    - "Question:"
  do_sample: false
  temperature: 0.0
29
repeats: 1
30
31
32
33
34
35
num_fewshot: 5
# filter_list:
#   - name: "get-answer"
#     filter:
#       - function: "regex"
#         regex_pattern: "### (\\-?[0-9\\.\\,]+)"
lintangsutawika's avatar
lintangsutawika committed
36
#       - function: "take_first"