mirror of
https://github.com/open-compass/opencompass.git
synced 2025-05-30 16:03:24 +08:00
[Fix] Fix rouge evaluator of rolebench_zh (#1322)
This commit is contained in:
parent
3aeabbc427
commit
0a1c89e618
@ -1,7 +1,7 @@
|
|||||||
from opencompass.openicl.icl_prompt_template import PromptTemplate
|
from opencompass.openicl.icl_prompt_template import PromptTemplate
|
||||||
from opencompass.openicl.icl_retriever import ZeroRetriever
|
from opencompass.openicl.icl_retriever import ZeroRetriever
|
||||||
from opencompass.openicl.icl_inferencer import GenInferencer
|
from opencompass.openicl.icl_inferencer import GenInferencer
|
||||||
from opencompass.openicl.icl_evaluator import RougeEvaluator
|
from opencompass.openicl.icl_evaluator import JiebaRougeEvaluator
|
||||||
from opencompass.datasets.rolebench import InstructionGeneralizationChineseDataset
|
from opencompass.datasets.rolebench import InstructionGeneralizationChineseDataset
|
||||||
|
|
||||||
instruction_generalization_zh_reader_cfg = dict(
|
instruction_generalization_zh_reader_cfg = dict(
|
||||||
@ -27,7 +27,7 @@ instruction_generalization_zh_infer_cfg = dict(
|
|||||||
)
|
)
|
||||||
|
|
||||||
instruction_generalization_zh_eval_cfg = dict(
|
instruction_generalization_zh_eval_cfg = dict(
|
||||||
evaluator=dict(type=RougeEvaluator),
|
evaluator=dict(type=JiebaRougeEvaluator),
|
||||||
pred_role='BOT'
|
pred_role='BOT'
|
||||||
)
|
)
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user