OpenCompass/configs/summarizers/groups/charm_rea.py
2024-05-27 16:17:01 +08:00

31 lines
832 B
Python

charm_tasks = [
'Anachronisms_Judgment',
'Movie_and_Music_Recommendation',
'Natural_Language_Inference',
'Reading_Comprehension',
'Sequence_Understanding',
'Sport_Understanding',
'Time_Understanding',
]
regions = [
'Chinese',
'Global',
]
prompts = [
'Direct',
'ZH-CoT',
'EN-CoT',
'XLT',
'Translate-EN',
]
charm_rea_summary_groups = []
for prompt in prompts:
for region in regions:
subsets = ['charm-rea-' + region + '_' + task + '_' + prompt for task in charm_tasks]
charm_rea_summary_groups.append({'name': 'charm-rea-' + region + '_' + prompt, 'subsets': subsets})
for prompt in prompts:
subsets = ['charm-rea-' + region + '_' + prompt for region in regions]
charm_rea_summary_groups.append({'name': 'charm-rea-' + prompt, 'subsets': subsets})