_template_yaml 862 Bytes
Newer Older
1
2
dataset_path : sean0042/KorMedMCQA
test_split : test
Gyouk Chu's avatar
Gyouk Chu committed
3
fewshot_split : fewshot
4
5
fewshot_config:
  sampler: first_n
Gyouk Chu's avatar
Gyouk Chu committed
6
7
  doc_to_text: "{{question.strip()}}\nA. {{A}}\nB. {{B}}\nC. {{C}}\nD. {{D}}\nE. {{E}}\n정답: {{['A', 'B', 'C', 'D', 'E'][answer-1]}}\n\n"
  doc_to_target: ""
8
9
10
11
12
13
14
15
16
17
output_type: generate_until
doc_to_text: "{{question.strip()}}\nA. {{A}}\nB. {{B}}\nC. {{C}}\nD. {{D}}\nE. {{E}}\n정답:"
doc_to_target: "{{['A', 'B', 'C', 'D', 'E'][answer-1]}}"
metric_list:
  - metric: exact_match
    aggregation: mean
    higher_is_better: true
    ignore_case: true
    ignore_punctuation: true
    regexes_to_ignore:
Gyouk Chu's avatar
Gyouk Chu committed
18
19
      - " "
      - "\n"
20
21
22
23
generation_kwargs:
  until:
    - "Q:"
    - "</s>"
Gyouk Chu's avatar
Gyouk Chu committed
24
    - "<|im_end|>"
25
    - "."
Gyouk Chu's avatar
Gyouk Chu committed
26
    - "\n\n"
27
28
  do_sample: false
  temperature: 0.0
Gyouk Chu's avatar
Gyouk Chu committed
29
30
31
32
33
  max_gen_toks: 1024
metadata:
  version: 2.0
dataset_kwargs:
  trust_remote_code: true