Commit 3e8135ce authored by Baber's avatar Baber
Browse files

Merge branch 'main' into comma

parents 8e560c96 0c134ee9
...@@ -24,7 +24,7 @@ metric_list: ...@@ -24,7 +24,7 @@ metric_list:
higher_is_better: true higher_is_better: true
num_fewshot: 4 num_fewshot: 4
metadata: metadata:
version: 2.0 version: 3.0
fewshot_config: fewshot_config:
sampler: first_n sampler: first_n
samples: !function utils.list_fewshot_samples samples: !function utils.list_fewshot_samples
...@@ -71,7 +71,7 @@ def list_fewshot_samples() -> list[dict]: ...@@ -71,7 +71,7 @@ def list_fewshot_samples() -> list[dict]:
] ]
def process_results(doc: dict, results: List[str]) -> Dict[str, int]: def process_results(doc: dict, results: list[str]) -> dict[str, int]:
candidates = results[0] candidates = results[0]
unnormalized_answer = get_unnormalized_answer(candidates) unnormalized_answer = get_unnormalized_answer(candidates)
...@@ -83,14 +83,17 @@ def process_results(doc: dict, results: List[str]) -> Dict[str, int]: ...@@ -83,14 +83,17 @@ def process_results(doc: dict, results: List[str]) -> Dict[str, int]:
retval = 0 retval = 0
# math_verify # math_verify
res = verify(parse(doc["answer"]), parse(candidates)) _mvres = verify(
mathval = 1 if res else 0 gold=parse(doc["solution"]),
target=parse(candidates),
)
mathval = 1 if _mvres else 0
results = { res = {
"exact_match": retval, "exact_match": retval,
"math_verify": mathval, "math_verify": mathval,
} }
return results return res
def last_boxed_only_string(string: str) -> Optional[str]: def last_boxed_only_string(string: str) -> Optional[str]:
......
...@@ -36,56 +36,56 @@ Homepage: `https://github.com/facebookresearch/MLQA` ...@@ -36,56 +36,56 @@ Homepage: `https://github.com/facebookresearch/MLQA`
#### Tasks #### Tasks
Tasks of the form `mlqa_context-lang_question-lang.yaml` Tasks of the form `mlqa_context-lang_question-lang`
* `mlqa_ar_ar.yaml` * `mlqa_ar_ar`
* `mlqa_ar_de.yaml` * `mlqa_ar_de`
* `mlqa_ar_vi.yaml` * `mlqa_ar_vi`
* `mlqa_ar_zh.yaml` * `mlqa_ar_zh`
* `mlqa_ar_en.yaml` * `mlqa_ar_en`
* `mlqa_ar_es.yaml` * `mlqa_ar_es`
* `mlqa_ar_hi.yaml` * `mlqa_ar_hi`
* `mlqa_de_ar.yaml` * `mlqa_de_ar`
* `mlqa_de_de.yaml` * `mlqa_de_de`
* `mlqa_de_vi.yaml` * `mlqa_de_vi`
* `mlqa_de_zh.yaml` * `mlqa_de_zh`
* `mlqa_de_en.yaml` * `mlqa_de_en`
* `mlqa_de_es.yaml` * `mlqa_de_es`
* `mlqa_de_hi.yaml` * `mlqa_de_hi`
* `mlqa_vi_ar.yaml` * `mlqa_vi_ar`
* `mlqa_vi_de.yaml` * `mlqa_vi_de`
* `mlqa_vi_vi.yaml` * `mlqa_vi_vi`
* `mlqa_vi_zh.yaml` * `mlqa_vi_zh`
* `mlqa_vi_en.yaml` * `mlqa_vi_en`
* `mlqa_vi_es.yaml` * `mlqa_vi_es`
* `mlqa_vi_hi.yaml` * `mlqa_vi_hi`
* `mlqa_zh_ar.yaml` * `mlqa_zh_ar`
* `mlqa_zh_de.yaml` * `mlqa_zh_de`
* `mlqa_zh_vi.yaml` * `mlqa_zh_vi`
* `mlqa_zh_zh.yaml` * `mlqa_zh_zh`
* `mlqa_zh_en.yaml` * `mlqa_zh_en`
* `mlqa_zh_es.yaml` * `mlqa_zh_es`
* `mlqa_zh_hi.yaml` * `mlqa_zh_hi`
* `mlqa_en_ar.yaml` * `mlqa_en_ar`
* `mlqa_en_de.yaml` * `mlqa_en_de`
* `mlqa_en_vi.yaml` * `mlqa_en_vi`
* `mlqa_en_zh.yaml` * `mlqa_en_zh`
* `mlqa_en_en.yaml` * `mlqa_en_en`
* `mlqa_en_es.yaml` * `mlqa_en_es`
* `mlqa_en_hi.yaml` * `mlqa_en_hi`
* `mlqa_es_ar.yaml` * `mlqa_es_ar`
* `mlqa_es_de.yaml` * `mlqa_es_de`
* `mlqa_es_vi.yaml` * `mlqa_es_vi`
* `mlqa_es_zh.yaml` * `mlqa_es_zh`
* `mlqa_es_en.yaml` * `mlqa_es_en`
* `mlqa_es_es.yaml` * `mlqa_es_es`
* `mlqa_es_hi.yaml` * `mlqa_es_hi`
* `mlqa_hi_ar.yaml` * `mlqa_hi_ar`
* `mlqa_hi_de.yaml` * `mlqa_hi_de`
* `mlqa_hi_vi.yaml` * `mlqa_hi_vi`
* `mlqa_hi_zh.yaml` * `mlqa_hi_zh`
* `mlqa_hi_en.yaml` * `mlqa_hi_en`
* `mlqa_hi_es.yaml` * `mlqa_hi_es`
* `mlqa_hi_hi.yaml` * `mlqa_hi_hi`
### Checklist ### Checklist
......
...@@ -71,3 +71,6 @@ switch to original implementation ...@@ -71,3 +71,6 @@ switch to original implementation
ver 2: PR #2116 ver 2: PR #2116
add missing newline in description. add missing newline in description.
PR #3137
Fix `mmlu_continuation` subgroup names to fit other variants, and switch dataset from `hails/mmlu_no_train` to `cais/mmlu` in all subtasks.
dataset_path: hails/mmlu_no_train # a copy of `cais/mmlu` with no auxiliary_train split dataset_path: cais/mmlu
output_type: multiple_choice output_type: multiple_choice
test_split: test test_split: test
fewshot_split: dev fewshot_split: dev
......
...@@ -3,25 +3,25 @@ group_alias: mmlu (continuation) ...@@ -3,25 +3,25 @@ group_alias: mmlu (continuation)
task: task:
- group: stem - group: stem
task: task:
- mmlu_continuation_stem - mmlu_stem_continuation
aggregate_metric_list: aggregate_metric_list:
- metric: acc - metric: acc
weight_by_size: True weight_by_size: True
- group: other - group: other
task: task:
- mmlu_continuation_other - mmlu_other_continuation
aggregate_metric_list: aggregate_metric_list:
- metric: acc - metric: acc
weight_by_size: True weight_by_size: True
- group: social sciences - group: social sciences
task: task:
- mmlu_continuation_social_sciences - mmlu_social_sciences_continuation
aggregate_metric_list: aggregate_metric_list:
- metric: acc - metric: acc
weight_by_size: True weight_by_size: True
- group: humanities - group: humanities
task: task:
- mmlu_continuation_humanities - mmlu_humanities_continuation
aggregate_metric_list: aggregate_metric_list:
- metric: acc - metric: acc
weight_by_size: True weight_by_size: True
......
"dataset_name": "abstract_algebra" "dataset_name": "abstract_algebra"
"description": "The following are questions (with answers) about abstract\ "description": "The following are questions (with answers) about abstract\
\ algebra.\n\n" \ algebra.\n\n"
"tag": "mmlu_continuation_stem" "tag": "mmlu_stem_continuation"
"include": "_continuation_template_yaml" "include": "_continuation_template_yaml"
"task": "mmlu_continuation_abstract_algebra" "task": "mmlu_abstract_algebra_continuation"
"dataset_name": "anatomy" "dataset_name": "anatomy"
"description": "The following are questions (with answers) about anatomy.\n\ "description": "The following are questions (with answers) about anatomy.\n\
\n" \n"
"tag": "mmlu_continuation_stem" "tag": "mmlu_stem_continuation"
"include": "_continuation_template_yaml" "include": "_continuation_template_yaml"
"task": "mmlu_continuation_anatomy" "task": "mmlu_anatomy_continuation"
"dataset_name": "astronomy" "dataset_name": "astronomy"
"description": "The following are questions (with answers) about astronomy.\n\ "description": "The following are questions (with answers) about astronomy.\n\
\n" \n"
"tag": "mmlu_continuation_stem" "tag": "mmlu_stem_continuation"
"include": "_continuation_template_yaml" "include": "_continuation_template_yaml"
"task": "mmlu_continuation_astronomy" "task": "mmlu_astronomy_continuation"
"dataset_name": "business_ethics" "dataset_name": "business_ethics"
"description": "The following are questions (with answers) about business\ "description": "The following are questions (with answers) about business\
\ ethics.\n\n" \ ethics.\n\n"
"tag": "mmlu_continuation_other" "tag": "mmlu_other_continuation"
"include": "_continuation_template_yaml" "include": "_continuation_template_yaml"
"task": "mmlu_continuation_business_ethics" "task": "mmlu_business_ethics_continuation"
"dataset_name": "clinical_knowledge" "dataset_name": "clinical_knowledge"
"description": "The following are questions (with answers) about clinical\ "description": "The following are questions (with answers) about clinical\
\ knowledge.\n\n" \ knowledge.\n\n"
"tag": "mmlu_continuation_other" "tag": "mmlu_other_continuation"
"include": "_continuation_template_yaml" "include": "_continuation_template_yaml"
"task": "mmlu_continuation_clinical_knowledge" "task": "mmlu_clinical_knowledge_continuation"
"dataset_name": "college_biology" "dataset_name": "college_biology"
"description": "The following are questions (with answers) about college\ "description": "The following are questions (with answers) about college\
\ biology.\n\n" \ biology.\n\n"
"tag": "mmlu_continuation_stem" "tag": "mmlu_stem_continuation"
"include": "_continuation_template_yaml" "include": "_continuation_template_yaml"
"task": "mmlu_continuation_college_biology" "task": "mmlu_college_biology_continuation"
"dataset_name": "college_chemistry" "dataset_name": "college_chemistry"
"description": "The following are questions (with answers) about college\ "description": "The following are questions (with answers) about college\
\ chemistry.\n\n" \ chemistry.\n\n"
"tag": "mmlu_continuation_stem" "tag": "mmlu_stem_continuation"
"include": "_continuation_template_yaml" "include": "_continuation_template_yaml"
"task": "mmlu_continuation_college_chemistry" "task": "mmlu_college_chemistry_continuation"
"dataset_name": "college_computer_science" "dataset_name": "college_computer_science"
"description": "The following are questions (with answers) about college\ "description": "The following are questions (with answers) about college\
\ computer science.\n\n" \ computer science.\n\n"
"tag": "mmlu_continuation_stem" "tag": "mmlu_stem_continuation"
"include": "_continuation_template_yaml" "include": "_continuation_template_yaml"
"task": "mmlu_continuation_college_computer_science" "task": "mmlu_college_computer_science_continuation"
"dataset_name": "college_mathematics" "dataset_name": "college_mathematics"
"description": "The following are questions (with answers) about college\ "description": "The following are questions (with answers) about college\
\ mathematics.\n\n" \ mathematics.\n\n"
"tag": "mmlu_continuation_stem" "tag": "mmlu_stem_continuation"
"include": "_continuation_template_yaml" "include": "_continuation_template_yaml"
"task": "mmlu_continuation_college_mathematics" "task": "mmlu_college_mathematics_continuation"
"dataset_name": "college_medicine" "dataset_name": "college_medicine"
"description": "The following are questions (with answers) about college\ "description": "The following are questions (with answers) about college\
\ medicine.\n\n" \ medicine.\n\n"
"tag": "mmlu_continuation_other" "tag": "mmlu_other_continuation"
"include": "_continuation_template_yaml" "include": "_continuation_template_yaml"
"task": "mmlu_continuation_college_medicine" "task": "mmlu_college_medicine_continuation"
"dataset_name": "college_physics" "dataset_name": "college_physics"
"description": "The following are questions (with answers) about college\ "description": "The following are questions (with answers) about college\
\ physics.\n\n" \ physics.\n\n"
"tag": "mmlu_continuation_stem" "tag": "mmlu_stem_continuation"
"include": "_continuation_template_yaml" "include": "_continuation_template_yaml"
"task": "mmlu_continuation_college_physics" "task": "mmlu_college_physics_continuation"
"dataset_name": "computer_security" "dataset_name": "computer_security"
"description": "The following are questions (with answers) about computer\ "description": "The following are questions (with answers) about computer\
\ security.\n\n" \ security.\n\n"
"tag": "mmlu_continuation_stem" "tag": "mmlu_stem_continuation"
"include": "_continuation_template_yaml" "include": "_continuation_template_yaml"
"task": "mmlu_continuation_computer_security" "task": "mmlu_computer_security_continuation"
"dataset_name": "conceptual_physics" "dataset_name": "conceptual_physics"
"description": "The following are questions (with answers) about conceptual\ "description": "The following are questions (with answers) about conceptual\
\ physics.\n\n" \ physics.\n\n"
"tag": "mmlu_continuation_stem" "tag": "mmlu_stem_continuation"
"include": "_continuation_template_yaml" "include": "_continuation_template_yaml"
"task": "mmlu_continuation_conceptual_physics" "task": "mmlu_conceptual_physics_continuation"
"dataset_name": "econometrics" "dataset_name": "econometrics"
"description": "The following are questions (with answers) about econometrics.\n\ "description": "The following are questions (with answers) about econometrics.\n\
\n" \n"
"tag": "mmlu_continuation_social_sciences" "tag": "mmlu_social_sciences_continuation"
"include": "_continuation_template_yaml" "include": "_continuation_template_yaml"
"task": "mmlu_continuation_econometrics" "task": "mmlu_econometrics_continuation"
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment