Skip to content

Commit a72c2f4

Browse files
committed
lighter mmlu
1 parent e0d2115 commit a72c2f4

File tree

2 files changed

+255
-254
lines changed

2 files changed

+255
-254
lines changed

src/tasksource/tasks.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -282,8 +282,9 @@ def _imppres_post_process(ds,prefix=''):
282282
)
283283

284284

285-
hendrycks_test = MultipleChoice('question',labels='answer',choices_list='choices',splits=['test','dev','validation'],
286-
config_name=get_dataset_config_names("hendrycks_test")
285+
mmlu = MultipleChoice('question',labels='answer',choices_list='choices',splits=['test','dev','validation'],
286+
dataset_name="tasksource/mmlu",
287+
config_name=get_dataset_config_names("tasksource/mmlu")
287288
)
288289

289290
winogrande = MultipleChoice('sentence',['option1','option2'],'answer',config_name='winogrande_xl',

0 commit comments

Comments
 (0)