File size: 442 Bytes
246d201 |
1 2 3 4 5 6 7 8 9 10 |
TASK_INFO_MAP = {
# === Reasoning ===
'gsm8k': {'class': 'ReasoningTask', 'type': 'reasoning'},
'math': {'class': 'ReasoningTask', 'type': 'reasoning'},
'mmlu': {'class': 'MultipleChoiceTask', 'type': 'reasoning'},
'theoremqa': {'class': 'TheoremqaTask', 'type': 'reasoning'},
'mbpp': {'class': 'MBPPTask', 'type': 'code_generation'},
'humaneval': {'class': 'HumanEvalTask', 'type': 'code_generation'},
}
|