mathqa.yaml 590 Bytes
Newer Older
haileyschoelkopf's avatar
haileyschoelkopf committed
1
2
group:
  - multiple_choice
haileyschoelkopf's avatar
haileyschoelkopf committed
3
  - math_word_problems
haileyschoelkopf's avatar
haileyschoelkopf committed
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
task: mathqa
dataset_path: math_qa
output_type: multiple_choice
training_split: train
validation_split: validation
test_split: test
create_choices: !function utils.create_choices # create list of answer choices
doc_to_text: "Question: {{Problem}}\nAnswer:"
doc_to_target: !function utils.doc_to_target
gold_alias: "{{['a', 'b', 'c', 'd', 'e'].index(correct)}}" # this will be cast to an int.
metric_list:
  - metric: acc
    aggregation: mean
    higher_is_better: true
  - metric: acc_norm
    aggregation: mean
    higher_is_better: true