소스 검색

adding mmlu to leaderboard confgs

Hamid Shojanazeri 1 년 전
부모
커밋
85c66acf21
2개의 변경된 파일11개의 추가작업 그리고 1개의 파일을 삭제
  1. 2 1
      eval/eval.py
  2. 9 0
      eval/open_llm_leaderboard/mmlu_5shots.yaml

+ 2 - 1
eval/eval.py

@@ -74,7 +74,8 @@ def load_tasks(args):
             "hellaswag_10_shot",
             "truthfulqa_mc2",
             "winogrande_5_shot",
-            "gsm8k"
+            "gsm8k",
+            "mmlu",
         ]
     return args.tasks.split(",") if args.tasks else []
         

+ 9 - 0
eval/open_llm_leaderboard/mmlu_5shots.yaml

@@ -0,0 +1,9 @@
+include: {$EVAL_PATH}/lm_eval/tasks/mmlu/default/_mmlu.yaml
+task:
+  - mmlu_stem
+  - mmlu_other
+  - mmlu_social_sciences
+  - mmlu_humanities
+num_fewshot: 5
+metric_list:
+  - metric: acc