math_agent.py 885 Bytes
Newer Older
Hubert's avatar
Hubert committed
1
2
3
4
5
from mmengine.config import read_base

with read_base():
    from .groups.mathbench_agent import mathbench_agent_summary_groups

Hubert's avatar
Hubert committed
6
7
8
9
10
11
12
13
14
15
16
17
18
summarizer = dict(
    dataset_abbrs=[
        '######## GSM8K-Agent Accuracy ########', # category
        ['gsm8k-agent', 'follow_acc'],
        ['gsm8k-agent', 'reasoning_acc'],
        ['gsm8k-agent', 'code_acc'],
        ['gsm8k-agent', 'action_pct'],
        '######## MATH-Agent Accuracy ########', # category
        ['math-agent', 'follow_acc'],
        ['math-agent', 'reasoning_acc'],
        ['math-agent', 'code_acc'],
        ['math-agent', 'action_pct'],
        '######## MathBench-Agent Accuracy ########', # category
Hubert's avatar
Hubert committed
19
20
21
        'mathbench-agent',
        'mathbench-circular-agent',
        'mathbench-circular-and-cloze-agent',
Hubert's avatar
Hubert committed
22
23
24
25
    ],
    summary_groups=sum(
        [v for k, v in locals().items() if k.endswith("_summary_groups")], [])
)