Commit 22b95671 authored by haileyschoelkopf's avatar haileyschoelkopf
Browse files

switch mmlu variants over to using

parent c1d9e625
...@@ -109,10 +109,9 @@ if __name__ == "__main__": ...@@ -109,10 +109,9 @@ if __name__ == "__main__":
yaml_dict = { yaml_dict = {
"include": base_yaml_name, "include": base_yaml_name,
"group": f"mmlu_{args.task_prefix}_{category}" "tag": f"mmlu_{args.task_prefix}_{category}"
if args.task_prefix != "" if args.task_prefix != ""
else f"mmlu_{category}", else f"mmlu_{category}",
"group_alias": category.replace("_", " "),
"task": f"mmlu_{args.task_prefix}_{subject}" "task": f"mmlu_{args.task_prefix}_{subject}"
if args.task_prefix != "" if args.task_prefix != ""
else f"mmlu_{subject}", else f"mmlu_{subject}",
......
"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"
"group": "mmlu_continuation_stem" "tag": "mmlu_continuation_stem"
"include": "_continuation_template_yaml" "include": "_continuation_template_yaml"
"task": "mmlu_continuation_abstract_algebra" "task": "mmlu_continuation_abstract_algebra"
"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"
"group": "mmlu_continuation_stem" "tag": "mmlu_continuation_stem"
"include": "_continuation_template_yaml" "include": "_continuation_template_yaml"
"task": "mmlu_continuation_anatomy" "task": "mmlu_continuation_anatomy"
"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"
"group": "mmlu_continuation_stem" "tag": "mmlu_continuation_stem"
"include": "_continuation_template_yaml" "include": "_continuation_template_yaml"
"task": "mmlu_continuation_astronomy" "task": "mmlu_continuation_astronomy"
"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"
"group": "mmlu_continuation_other" "tag": "mmlu_continuation_other"
"include": "_continuation_template_yaml" "include": "_continuation_template_yaml"
"task": "mmlu_continuation_business_ethics" "task": "mmlu_continuation_business_ethics"
"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"
"group": "mmlu_continuation_other" "tag": "mmlu_continuation_other"
"include": "_continuation_template_yaml" "include": "_continuation_template_yaml"
"task": "mmlu_continuation_clinical_knowledge" "task": "mmlu_continuation_clinical_knowledge"
"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"
"group": "mmlu_continuation_stem" "tag": "mmlu_continuation_stem"
"include": "_continuation_template_yaml" "include": "_continuation_template_yaml"
"task": "mmlu_continuation_college_biology" "task": "mmlu_continuation_college_biology"
"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"
"group": "mmlu_continuation_stem" "tag": "mmlu_continuation_stem"
"include": "_continuation_template_yaml" "include": "_continuation_template_yaml"
"task": "mmlu_continuation_college_chemistry" "task": "mmlu_continuation_college_chemistry"
"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"
"group": "mmlu_continuation_stem" "tag": "mmlu_continuation_stem"
"include": "_continuation_template_yaml" "include": "_continuation_template_yaml"
"task": "mmlu_continuation_college_computer_science" "task": "mmlu_continuation_college_computer_science"
"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"
"group": "mmlu_continuation_stem" "tag": "mmlu_continuation_stem"
"include": "_continuation_template_yaml" "include": "_continuation_template_yaml"
"task": "mmlu_continuation_college_mathematics" "task": "mmlu_continuation_college_mathematics"
"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"
"group": "mmlu_continuation_other" "tag": "mmlu_continuation_other"
"include": "_continuation_template_yaml" "include": "_continuation_template_yaml"
"task": "mmlu_continuation_college_medicine" "task": "mmlu_continuation_college_medicine"
"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"
"group": "mmlu_continuation_stem" "tag": "mmlu_continuation_stem"
"include": "_continuation_template_yaml" "include": "_continuation_template_yaml"
"task": "mmlu_continuation_college_physics" "task": "mmlu_continuation_college_physics"
"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"
"group": "mmlu_continuation_stem" "tag": "mmlu_continuation_stem"
"include": "_continuation_template_yaml" "include": "_continuation_template_yaml"
"task": "mmlu_continuation_computer_security" "task": "mmlu_continuation_computer_security"
"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"
"group": "mmlu_continuation_stem" "tag": "mmlu_continuation_stem"
"include": "_continuation_template_yaml" "include": "_continuation_template_yaml"
"task": "mmlu_continuation_conceptual_physics" "task": "mmlu_continuation_conceptual_physics"
"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"
"group": "mmlu_continuation_social_sciences" "tag": "mmlu_continuation_social_sciences"
"include": "_continuation_template_yaml" "include": "_continuation_template_yaml"
"task": "mmlu_continuation_econometrics" "task": "mmlu_continuation_econometrics"
"dataset_name": "electrical_engineering" "dataset_name": "electrical_engineering"
"description": "The following are questions (with answers) about electrical\ "description": "The following are questions (with answers) about electrical\
\ engineering.\n\n" \ engineering.\n\n"
"group": "mmlu_continuation_stem" "tag": "mmlu_continuation_stem"
"include": "_continuation_template_yaml" "include": "_continuation_template_yaml"
"task": "mmlu_continuation_electrical_engineering" "task": "mmlu_continuation_electrical_engineering"
"dataset_name": "elementary_mathematics" "dataset_name": "elementary_mathematics"
"description": "The following are questions (with answers) about elementary\ "description": "The following are questions (with answers) about elementary\
\ mathematics.\n\n" \ mathematics.\n\n"
"group": "mmlu_continuation_stem" "tag": "mmlu_continuation_stem"
"include": "_continuation_template_yaml" "include": "_continuation_template_yaml"
"task": "mmlu_continuation_elementary_mathematics" "task": "mmlu_continuation_elementary_mathematics"
"dataset_name": "formal_logic" "dataset_name": "formal_logic"
"description": "The following are questions (with answers) about formal\ "description": "The following are questions (with answers) about formal\
\ logic.\n\n" \ logic.\n\n"
"group": "mmlu_continuation_humanities" "tag": "mmlu_continuation_humanities"
"include": "_continuation_template_yaml" "include": "_continuation_template_yaml"
"task": "mmlu_continuation_formal_logic" "task": "mmlu_continuation_formal_logic"
"dataset_name": "global_facts" "dataset_name": "global_facts"
"description": "The following are questions (with answers) about global\ "description": "The following are questions (with answers) about global\
\ facts.\n\n" \ facts.\n\n"
"group": "mmlu_continuation_other" "tag": "mmlu_continuation_other"
"include": "_continuation_template_yaml" "include": "_continuation_template_yaml"
"task": "mmlu_continuation_global_facts" "task": "mmlu_continuation_global_facts"
"dataset_name": "high_school_biology" "dataset_name": "high_school_biology"
"description": "The following are questions (with answers) about high\ "description": "The following are questions (with answers) about high\
\ school biology.\n\n" \ school biology.\n\n"
"group": "mmlu_continuation_stem" "tag": "mmlu_continuation_stem"
"include": "_continuation_template_yaml" "include": "_continuation_template_yaml"
"task": "mmlu_continuation_high_school_biology" "task": "mmlu_continuation_high_school_biology"
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