OpenCompass/opencompass/configs/summarizers/bbeh.py

30 lines
883 B
Python
Raw Normal View History

2025-03-16 11:25:32 +08:00
from mmengine.config import read_base
with read_base():
from .groups.bbeh import bbeh_summary_groups
2025-03-17 16:46:44 +08:00
# Get all the BBEH subset names from the imported bbeh_summary_groups
bbeh_subsets = []
for group in bbeh_summary_groups:
if group['name'] == 'bbeh':
bbeh_subsets = group['subsets']
break
2025-03-16 11:25:32 +08:00
summarizer = dict(
2025-03-17 16:46:44 +08:00
# Include both individual datasets and the summary metrics we want to see
2025-03-17 16:57:35 +08:00
dataset_abbrs=bbeh_subsets + ['bbeh_naive_average'] + ['bbeh_harmonic_mean'],
2025-03-17 16:46:44 +08:00
# Define the summary group for bbeh
summary_groups=[
{
2025-03-17 16:57:35 +08:00
'name': 'bbeh_naive_average',
2025-03-17 16:46:44 +08:00
'subsets': bbeh_subsets,
2025-03-17 16:57:35 +08:00
'metric': 'naive_average' # Explicitly specify the metric to use
2025-03-17 16:46:44 +08:00
},
{
'name': 'bbeh_harmonic_mean',
'subsets': bbeh_subsets,
'metric': 'harmonic_mean'
}
]
2025-03-16 11:25:32 +08:00
)