OpenCompass/opencompass/configs/datasets/OpenHuEval/HuLifeQA/HuLifeQA.py

52 lines
1.3 KiB
Python
Raw Normal View History

2025-01-24 10:32:17 +08:00
from opencompass.datasets import WildBenchDataset
from opencompass.openicl.icl_evaluator import LMEvaluator
from opencompass.openicl.icl_inferencer import ChatInferencer
from opencompass.openicl.icl_prompt_template import PromptTemplate
from opencompass.openicl.icl_retriever import ZeroRetriever
2025-03-12 16:15:46 +08:00
with read_base():
from .HuLifeQA_setting import DATA_PATH, TASK_GROUP_NEW
2025-01-24 10:32:17 +08:00
hu_life_qa_reader_cfg = dict(
2025-02-10 09:38:49 +08:00
input_columns=['dialogue', 'prompt'],
output_column='judge',
2025-01-24 10:32:17 +08:00
)
hu_life_qa_infer_cfg = dict(
prompt_template=dict(
type=PromptTemplate,
template="""{dialogue}"""
),
retriever=dict(type=ZeroRetriever),
inferencer=dict(
type=ChatInferencer,
max_seq_len=8192,
max_out_len=8192,
2025-02-10 09:38:49 +08:00
infer_mode='last',
2025-01-24 10:32:17 +08:00
),
)
hu_life_qa_eval_cfg = dict(
evaluator=dict(
type=LMEvaluator,
prompt_template=dict(
type=PromptTemplate,
template="""{prompt}"""
),
),
2025-02-10 09:38:49 +08:00
pred_role='BOT',
2025-01-24 10:32:17 +08:00
)
2025-03-12 16:15:46 +08:00
hu_life_qa_datasets = []
2025-01-24 10:32:17 +08:00
hu_life_qa_datasets.append(
dict(
2025-02-10 09:38:49 +08:00
abbr='open_hu_eval_hu_life_qa',
2025-01-24 10:32:17 +08:00
type=WildBenchDataset,
2025-03-12 16:15:46 +08:00
path=DATA_PATH,
2025-01-24 10:32:17 +08:00
reader_cfg=hu_life_qa_reader_cfg,
infer_cfg=hu_life_qa_infer_cfg,
eval_cfg=hu_life_qa_eval_cfg,
)
)