mathvista.yaml 666 Bytes
Newer Older
Baber's avatar
Baber committed
1
dataset_path: AI4Math/MathVista
Baber's avatar
Baber committed
2
task: mathvista
Baber's avatar
Baber committed
3
test_split: testmini
Baber's avatar
Baber committed
4
5
output_type: "generate_until"
#process_docs: !function utils.process_docs
Baber's avatar
Baber committed
6
7
doc_to_image:
  - decoded_image
Baber's avatar
Baber committed
8
doc_to_text: "<image>{{query}}"
Baber's avatar
Baber committed
9
10
doc_to_target: answer
process_results: !function utils.process_results
Baber's avatar
Baber committed
11
12
13
14
15
generation_kwargs:
  until:
    - "<|endoftext|>"
  temperature: 0.0
  do_sample: false
Baber's avatar
Baber committed
16
  max_gen_toks: 1024
Baber's avatar
nits  
Baber committed
17
18
19
20
filter_list:
  - name: "extract_answer"
    filter:
      - function: "custom"
Baber's avatar
fix  
Baber committed
21
        filter_fn: !function utils.extract_all_answers
Baber's avatar
Baber committed
22
23
24
25
26
27
28
29
metric_list:
  - metric: acc
    aggregation: mean
    higher_is_better: true
metadata:
  version: 1.0
dataset_kwargs:
  trust_remote_code: true