update flan mmlu

lm_eval/tasks/mmlu/_generate_configs.py

@@ -106,7 +106,7 @@ if __name__ == "__main__":
 
         yaml_dict = {
             "include": base_yaml_name,
-            "task": f"mmlu_{args.task_prefix}_{subject}",
+            "task": f"mmlu_{args.task_prefix}_{subject}" if args.task_prefix != "" else f"mmlu_{subject}",
             "dataset_name": subject,
             "description": description,
         }

lm_eval/tasks/mmlu/default/_default_template_yaml

+group: mmlu
+dataset_path: cais/mmlu
+test_split: test
+fewshot_split: dev
+output_type: multiple_choice
+doc_to_text: "{{question.strip()}}\nA. {{choices[0]}}\nB. {{choices[1]}}\nC. {{choices[2]}}\nD. {{choices[3]}}\nAnswer:"
+doc_to_choice: ["A", "B", "C", "D"]
+doc_to_target: answer
+metric_list:
+  - metric: acc
+    aggregation: mean
+    higher_is_better: true
+  - metric: acc_norm
+    aggregation: mean
+    higher_is_better: true
\ No newline at end of file

lm_eval/tasks/mmlu/default/mmlu_abstract_algebra.yaml

+"dataset_name": "abstract_algebra"
+"description": "The following are multiple choice questions (with answers) about abstract algebra.\n\n"
+"include": "_default_template_yaml"
+"task": "mmlu_abstract_algebra"

lm_eval/tasks/mmlu/default/mmlu_anatomy.yaml

+"dataset_name": "anatomy"
+"description": "The following are multiple choice questions (with answers) about anatomy.\n\n"
+"include": "_default_template_yaml"
+"task": "mmlu_anatomy"

lm_eval/tasks/mmlu/default/mmlu_astronomy.yaml

+"dataset_name": "astronomy"
+"description": "The following are multiple choice questions (with answers) about astronomy.\n\n"
+"include": "_default_template_yaml"
+"task": "mmlu_astronomy"

lm_eval/tasks/mmlu/default/mmlu_business_ethics.yaml

+"dataset_name": "business_ethics"
+"description": "The following are multiple choice questions (with answers) about business ethics.\n\n"
+"include": "_default_template_yaml"
+"task": "mmlu_business_ethics"

lm_eval/tasks/mmlu/default/mmlu_clinical_knowledge.yaml

+"dataset_name": "clinical_knowledge"
+"description": "The following are multiple choice questions (with answers) about clinical knowledge.\n\n"
+"include": "_default_template_yaml"
+"task": "mmlu_clinical_knowledge"

lm_eval/tasks/mmlu/default/mmlu_college_biology.yaml

+"dataset_name": "college_biology"
+"description": "The following are multiple choice questions (with answers) about college biology.\n\n"
+"include": "_default_template_yaml"
+"task": "mmlu_college_biology"

lm_eval/tasks/mmlu/default/mmlu_college_chemistry.yaml

+"dataset_name": "college_chemistry"
+"description": "The following are multiple choice questions (with answers) about college chemistry.\n\n"
+"include": "_default_template_yaml"
+"task": "mmlu_college_chemistry"

lm_eval/tasks/mmlu/default/mmlu_college_computer_science.yaml

+"dataset_name": "college_computer_science"
+"description": "The following are multiple choice questions (with answers) about college computer science.\n\n"
+"include": "_default_template_yaml"
+"task": "mmlu_college_computer_science"

lm_eval/tasks/mmlu/default/mmlu_college_mathematics.yaml

+"dataset_name": "college_mathematics"
+"description": "The following are multiple choice questions (with answers) about college mathematics.\n\n"
+"include": "_default_template_yaml"
+"task": "mmlu_college_mathematics"

lm_eval/tasks/mmlu/default/mmlu_college_medicine.yaml

+"dataset_name": "college_medicine"
+"description": "The following are multiple choice questions (with answers) about college medicine.\n\n"
+"include": "_default_template_yaml"
+"task": "mmlu_college_medicine"

lm_eval/tasks/mmlu/default/mmlu_college_physics.yaml

+"dataset_name": "college_physics"
+"description": "The following are multiple choice questions (with answers) about college physics.\n\n"
+"include": "_default_template_yaml"
+"task": "mmlu_college_physics"

lm_eval/tasks/mmlu/default/mmlu_computer_security.yaml

+"dataset_name": "computer_security"
+"description": "The following are multiple choice questions (with answers) about computer security.\n\n"
+"include": "_default_template_yaml"
+"task": "mmlu_computer_security"

lm_eval/tasks/mmlu/default/mmlu_conceptual_physics.yaml

+"dataset_name": "conceptual_physics"
+"description": "The following are multiple choice questions (with answers) about conceptual physics.\n\n"
+"include": "_default_template_yaml"
+"task": "mmlu_conceptual_physics"

lm_eval/tasks/mmlu/default/mmlu_econometrics.yaml

+"dataset_name": "econometrics"
+"description": "The following are multiple choice questions (with answers) about econometrics.\n\n"
+"include": "_default_template_yaml"
+"task": "mmlu_econometrics"

lm_eval/tasks/mmlu/default/mmlu_electrical_engineering.yaml

+"dataset_name": "electrical_engineering"
+"description": "The following are multiple choice questions (with answers) about electrical engineering.\n\n"
+"include": "_default_template_yaml"
+"task": "mmlu_electrical_engineering"

lm_eval/tasks/mmlu/default/mmlu_elementary_mathematics.yaml

+"dataset_name": "elementary_mathematics"
+"description": "The following are multiple choice questions (with answers) about elementary mathematics.\n\n"
+"include": "_default_template_yaml"
+"task": "mmlu_elementary_mathematics"

lm_eval/tasks/mmlu/default/mmlu_formal_logic.yaml

+"dataset_name": "formal_logic"
+"description": "The following are multiple choice questions (with answers) about formal logic.\n\n"
+"include": "_default_template_yaml"
+"task": "mmlu_formal_logic"

lm_eval/tasks/mmlu/default/mmlu_global_facts.yaml

+"dataset_name": "global_facts"
+"description": "The following are multiple choice questions (with answers) about global facts.\n\n"
+"include": "_default_template_yaml"
+"task": "mmlu_global_facts"