gsm8k.yaml 586 Bytes
Newer Older
Lintang Sutawika's avatar
Lintang Sutawika committed
1
tag:
2
  - math_word_problems
3
task: gsm8k
4
5
dataset_path: gsm8k
dataset_name: main
6
output_type: generate_until
7
8
9
training_split: train
fewshot_split: train
test_split: test
Baber's avatar
Baber committed
10
process_results: !function utils.process_results
11
12
13
doc_to_text: "Question: {{question}}\nAnswer:"
doc_to_target: "{{answer}}" #" {{answer.split('### ')[-1].rstrip()}}"
metric_list:
Baber's avatar
Baber committed
14
  - metric: math_verify
15
16
17
18
19
    aggregation: mean
    higher_is_better: true
generation_kwargs:
  until:
    - "Question:"
20
21
    - "</s>"
    - "<|im_end|>"
22
  do_sample: false
23
  temperature: 0.0
24
repeats: 1
25
num_fewshot: 5
lintangsutawika's avatar
lintangsutawika committed
26
metadata:
27
  version: 3.0