afrimgsm_common_yaml 1.07 KB
Newer Older
JessicaOjo's avatar
JessicaOjo committed
1
2
3
group:
 - mgsm_direct
 - afrimgsm
4
5
6
7
8
9
10
11
12
dataset_path: masakhane/afrimgsm
output_type: generate_until
training_split: train
test_split: test
fewshot_split: train
target_delimiter: ""
doc_to_target: '{% if answer is not none %}{{answer}}{% else %}{{answer_number|string}}{% endif %}'
doc_to_text: '{% if answer is not none %}{{question+"\nAnswer:"}}{% else %}{{"Question: "+question+"\nAnswer:"}}{% endif %}'
generation_kwargs:
JessicaOjo's avatar
JessicaOjo committed
13
  do_sample: false
JessicaOjo's avatar
JessicaOjo committed
14
15
16
17
  until:
  - 'Question:'
  - </s>
  - <|im_end|>
18
filter_list:
JessicaOjo's avatar
JessicaOjo committed
19
20
21
22
  - name: remove_whitespace
    filter:
      - function: remove_whitespace
      - function: take_first
23
  - filter:
JessicaOjo's avatar
JessicaOjo committed
24
    - function: regex
25
      group_select: -1
JessicaOjo's avatar
JessicaOjo committed
26
      regex_pattern: (-?[$0-9.,]{2,})|(-?[0-9]+)
27
28
29
    - function: take_first
    name: flexible-extract
metric_list:
JessicaOjo's avatar
JessicaOjo committed
30
31
32
33
34
  - metric: exact_match
    aggregation: mean
    higher_is_better: true
    ignore_case: true
    ignore_punctuation: true
JessicaOjo's avatar
JessicaOjo committed
35
36
37
38
39
40
  - metric: f1
    aggregation: !function utils.weighted_f1_score
    average: weighted
    higher_is_better: True
    ignore_case: true
    ignore_punctuation: true
41
metadata:
JessicaOjo's avatar
JessicaOjo committed
42
  version: 2.0