|
|
|
import csv |
|
import os |
|
import os.path as osp |
|
import re |
|
from collections import defaultdict |
|
from datetime import datetime |
|
|
|
import numpy as np |
|
from mmengine import ConfigDict |
|
|
|
try: |
|
from prettytable import from_csv |
|
except ImportError: |
|
from_csv = None |
|
|
|
from opencompass.utils import model_abbr_from_cfg |
|
|
|
from .alignmentbench import AlignmentBenchSummarizer, post_process_alignbench |
|
from .subjective_post_process import post_process_autoj, post_process_judgelm |
|
from .utils import get_judgeanswer_and_reference, get_outdir |
|
|
|
CATEGORIES = { |
|
'中文': ['内容扩写_ZH', '内容续写_ZH', '内容改写_ZH'], |
|
'英文': ['内容扩写_EN', '内容续写_EN', '内容改写_EN'], |
|
} |
|
|
|
All_Dimensions = [ |
|
'Creativity', 'Richness', 'User Demand Fulfillment', 'Logical Coherence', |
|
'Overall Score', '创造性', '丰富度', '满足用户需求', '逻辑连贯性', '综合得分' |
|
] |
|
|
|
|
|
def post_process_creationbench(judgement: str, |
|
all_dimensions=All_Dimensions, |
|
possible_keys=['综合得分', 'Overall Score']): |
|
"""Input a string like below: |
|
|
|
xxx{'事实正确性': 1, '满足用户需求': 1, '清晰度': 2, '完备性': 1, '综合得分': 1}xxx, |
|
and extract each score |
|
""" |
|
return post_process_alignbench(judgement, all_dimensions, possible_keys) |
|
|
|
|
|
class CreationBenchSummarizer(AlignmentBenchSummarizer): |
|
"""Do the subjectivity analyze based on evaluation results. |
|
|
|
Args: |
|
config (ConfigDict): The configuration object of the evaluation task. |
|
It's expected to be filled out at runtime. |
|
""" |
|
|
|
def __init__(self, config: ConfigDict, judge_type: str) -> None: |
|
super().__init__(config, judge_type) |
|
self.judge_map = { |
|
'general': post_process_creationbench, |
|
'autoj': post_process_autoj, |
|
'judgelm': post_process_judgelm |
|
} |
|
self.judge_function = self.judge_map[self.judge_type] |
|
self.category = CATEGORIES |
|
|
|
def summarize(self, |
|
time_str: str = datetime.now().strftime('%Y%m%d_%H%M%S')): |
|
"""Summarize the subjectivity analysis based on evaluation results. |
|
|
|
Args: |
|
time_str (str): Timestamp for file naming. |
|
|
|
Returns: |
|
pd.DataFrame: The summary results. |
|
""" |
|
super().summarize(time_str) |
|
|