Unverified Commit 86d5ec3d authored by Leymore's avatar Leymore Committed by GitHub
Browse files

Update configs (#9)

* Update implements

* Update
parent 2d0b184b
...@@ -3,6 +3,7 @@ from opencompass.openicl.icl_retriever import ZeroRetriever ...@@ -3,6 +3,7 @@ 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 AccEvaluator from opencompass.openicl.icl_evaluator import AccEvaluator
from opencompass.datasets import ARCDataset from opencompass.datasets import ARCDataset
from opencompass.utils.text_postprocessors import first_capital_postprocess
ARC_c_reader_cfg = dict( ARC_c_reader_cfg = dict(
input_columns=["question", "textA", "textB", "textC", "textD"], input_columns=["question", "textA", "textB", "textC", "textD"],
...@@ -27,7 +28,7 @@ ARC_c_infer_cfg = dict( ...@@ -27,7 +28,7 @@ ARC_c_infer_cfg = dict(
ARC_c_eval_cfg = dict( ARC_c_eval_cfg = dict(
evaluator=dict(type=AccEvaluator), evaluator=dict(type=AccEvaluator),
pred_role="BOT", pred_role="BOT",
pred_postprocessor=dict(type="first-capital"), pred_postprocessor=dict(type=first_capital_postprocess),
) )
ARC_c_datasets = [ ARC_c_datasets = [
......
...@@ -3,6 +3,7 @@ from opencompass.openicl.icl_retriever import ZeroRetriever ...@@ -3,6 +3,7 @@ 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 AccEvaluator from opencompass.openicl.icl_evaluator import AccEvaluator
from opencompass.datasets import ARCDataset from opencompass.datasets import ARCDataset
from opencompass.utils.text_postprocessors import first_capital_postprocess
ARC_e_reader_cfg = dict( ARC_e_reader_cfg = dict(
input_columns=["question", "textA", "textB", "textC", "textD"], input_columns=["question", "textA", "textB", "textC", "textD"],
...@@ -27,7 +28,7 @@ ARC_e_infer_cfg = dict( ...@@ -27,7 +28,7 @@ ARC_e_infer_cfg = dict(
ARC_e_eval_cfg = dict( ARC_e_eval_cfg = dict(
evaluator=dict(type=AccEvaluator), evaluator=dict(type=AccEvaluator),
pred_role="BOT", pred_role="BOT",
pred_postprocessor=dict(type="first-capital"), pred_postprocessor=dict(type=first_capital_postprocess),
) )
ARC_e_datasets = [ ARC_e_datasets = [
......
...@@ -3,6 +3,7 @@ from opencompass.openicl.icl_retriever import ZeroRetriever ...@@ -3,6 +3,7 @@ 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 AccEvaluator from opencompass.openicl.icl_evaluator import AccEvaluator
from opencompass.datasets import C3Dataset_V2 from opencompass.datasets import C3Dataset_V2
from opencompass.utils.text_postprocessors import first_capital_postprocess
C3_reader_cfg = dict( C3_reader_cfg = dict(
input_columns=[ input_columns=[
...@@ -35,7 +36,7 @@ C3_infer_cfg = dict( ...@@ -35,7 +36,7 @@ C3_infer_cfg = dict(
C3_eval_cfg = dict( C3_eval_cfg = dict(
evaluator=dict(type=AccEvaluator), evaluator=dict(type=AccEvaluator),
pred_role="BOT", pred_role="BOT",
pred_postprocessor=dict(type="first-capital"), pred_postprocessor=dict(type=first_capital_postprocess),
) )
C3_datasets = [ C3_datasets = [
......
from mmengine.config import read_base from mmengine.config import read_base
with read_base(): with read_base():
from .CLUE_CMRC_gen_941108 import CMRC_datasets # noqa: F401, F403 from .CLUE_CMRC_gen_1bd3c8 import CMRC_datasets # noqa: F401, F403
from opencompass.openicl.icl_prompt_template import PromptTemplate
from opencompass.openicl.icl_retriever import ZeroRetriever
from opencompass.openicl.icl_inferencer import GenInferencer
from opencompass.openicl.icl_evaluator import EMEvaluator
from opencompass.datasets import CMRCDataset, cmrc_postprocess
CMRC_reader_cfg = dict(
input_columns=['question', 'context'], output_column='answers')
CMRC_infer_cfg = dict(
prompt_template=dict(
type=PromptTemplate,
template=dict(round=[
dict(
role="HUMAN",
prompt="根据文章回答问题。你的答案应该尽可能简练,请以 ‘答案是’ 开头的句式作答。\n文章:{context}\n问:{question}\n答:"),
])),
retriever=dict(type=ZeroRetriever),
inferencer=dict(type=GenInferencer))
CMRC_eval_cfg = dict(
evaluator=dict(type=EMEvaluator),
pred_role="BOT",
pred_postprocessor=dict(type=cmrc_postprocess),
)
CMRC_datasets = [
dict(
type=CMRCDataset,
abbr='CMRC_dev',
path='./data/CLUE/CMRC/dev.json',
reader_cfg=CMRC_reader_cfg,
infer_cfg=CMRC_infer_cfg,
eval_cfg=CMRC_eval_cfg),
]
from mmengine.config import read_base from mmengine.config import read_base
with read_base(): with read_base():
from .CLUE_DRCD_gen_941108 import DRCD_datasets # noqa: F401, F403 from .CLUE_DRCD_gen_1bd3c8 import DRCD_datasets # noqa: F401, F403
from opencompass.openicl.icl_prompt_template import PromptTemplate
from opencompass.openicl.icl_retriever import ZeroRetriever
from opencompass.openicl.icl_inferencer import GenInferencer
from opencompass.openicl.icl_evaluator import EMEvaluator
from opencompass.datasets import DRCDDataset, drcd_postprocess
DRCD_reader_cfg = dict(
input_columns=['question', 'context'], output_column='answers')
DRCD_infer_cfg = dict(
prompt_template=dict(
type=PromptTemplate,
template=dict(round=[
dict(
role="HUMAN",
prompt="根据文章回答问题。你的答案应该尽可能简练,请以 ‘答案是’ 开头的句式作答。\n文章:{context}\n问:{question}\n答:"),
])),
retriever=dict(type=ZeroRetriever),
inferencer=dict(type=GenInferencer))
DRCD_eval_cfg = dict(
evaluator=dict(type=EMEvaluator),
pred_role="BOT",
pred_postprocessor=dict(type=drcd_postprocess),
)
DRCD_datasets = [
dict(
type=DRCDDataset,
abbr='DRCD_dev',
path='./data/CLUE/DRCD/dev.json',
reader_cfg=DRCD_reader_cfg,
infer_cfg=DRCD_infer_cfg,
eval_cfg=DRCD_eval_cfg),
]
...@@ -3,6 +3,7 @@ from opencompass.openicl.icl_retriever import ZeroRetriever ...@@ -3,6 +3,7 @@ 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 AccEvaluator from opencompass.openicl.icl_evaluator import AccEvaluator
from opencompass.datasets import AFQMCDataset_V2 from opencompass.datasets import AFQMCDataset_V2
from opencompass.utils.text_postprocessors import first_capital_postprocess
afqmc_reader_cfg = dict( afqmc_reader_cfg = dict(
input_columns=["sentence1", "sentence2"], input_columns=["sentence1", "sentence2"],
...@@ -27,7 +28,7 @@ afqmc_infer_cfg = dict( ...@@ -27,7 +28,7 @@ afqmc_infer_cfg = dict(
afqmc_eval_cfg = dict( afqmc_eval_cfg = dict(
evaluator=dict(type=AccEvaluator), evaluator=dict(type=AccEvaluator),
pred_role="BOT", pred_role="BOT",
pred_postprocessor=dict(type="first-capital"), pred_postprocessor=dict(type=first_capital_postprocess),
) )
afqmc_datasets = [ afqmc_datasets = [
......
...@@ -3,6 +3,7 @@ from opencompass.openicl.icl_retriever import ZeroRetriever ...@@ -3,6 +3,7 @@ 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 AccEvaluator from opencompass.openicl.icl_evaluator import AccEvaluator
from opencompass.datasets import cmnliDataset_V2 from opencompass.datasets import cmnliDataset_V2
from opencompass.utils.text_postprocessors import first_capital_postprocess
cmnli_reader_cfg = dict( cmnli_reader_cfg = dict(
input_columns=["sentence1", "sentence2"], input_columns=["sentence1", "sentence2"],
...@@ -27,7 +28,7 @@ cmnli_infer_cfg = dict( ...@@ -27,7 +28,7 @@ cmnli_infer_cfg = dict(
cmnli_eval_cfg = dict( cmnli_eval_cfg = dict(
evaluator=dict(type=AccEvaluator), evaluator=dict(type=AccEvaluator),
pred_role="BOT", pred_role="BOT",
pred_postprocessor=dict(type="first-capital"), pred_postprocessor=dict(type=first_capital_postprocess),
) )
cmnli_datasets = [ cmnli_datasets = [
......
...@@ -3,6 +3,7 @@ from opencompass.openicl.icl_retriever import ZeroRetriever ...@@ -3,6 +3,7 @@ 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 AccEvaluator from opencompass.openicl.icl_evaluator import AccEvaluator
from opencompass.datasets import cmnliDataset_V2 from opencompass.datasets import cmnliDataset_V2
from opencompass.utils.text_postprocessors import first_capital_postprocess
cmnli_reader_cfg = dict( cmnli_reader_cfg = dict(
input_columns=["sentence1", "sentence2"], input_columns=["sentence1", "sentence2"],
...@@ -27,7 +28,7 @@ cmnli_infer_cfg = dict( ...@@ -27,7 +28,7 @@ cmnli_infer_cfg = dict(
cmnli_eval_cfg = dict( cmnli_eval_cfg = dict(
evaluator=dict(type=AccEvaluator), evaluator=dict(type=AccEvaluator),
pred_role="BOT", pred_role="BOT",
pred_postprocessor=dict(type="first-capital"), pred_postprocessor=dict(type=first_capital_postprocess),
) )
cmnli_datasets = [ cmnli_datasets = [
......
...@@ -3,6 +3,7 @@ from opencompass.openicl.icl_retriever import ZeroRetriever ...@@ -3,6 +3,7 @@ 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 AccEvaluator from opencompass.openicl.icl_evaluator import AccEvaluator
from opencompass.datasets import cmnliDataset_V2 from opencompass.datasets import cmnliDataset_V2
from opencompass.utils.text_postprocessors import first_capital_postprocess
ocnli_reader_cfg = dict( ocnli_reader_cfg = dict(
input_columns=["sentence1", "sentence2"], input_columns=["sentence1", "sentence2"],
...@@ -28,7 +29,7 @@ ocnli_infer_cfg = dict( ...@@ -28,7 +29,7 @@ ocnli_infer_cfg = dict(
ocnli_eval_cfg = dict( ocnli_eval_cfg = dict(
evaluator=dict(type=AccEvaluator), evaluator=dict(type=AccEvaluator),
pred_role="BOT", pred_role="BOT",
pred_postprocessor=dict(type="first-capital"), pred_postprocessor=dict(type=first_capital_postprocess),
) )
ocnli_datasets = [ ocnli_datasets = [
......
...@@ -3,6 +3,7 @@ from opencompass.openicl.icl_retriever import ZeroRetriever ...@@ -3,6 +3,7 @@ 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 AccEvaluator from opencompass.openicl.icl_evaluator import AccEvaluator
from opencompass.datasets import cmnliDataset_V2 from opencompass.datasets import cmnliDataset_V2
from opencompass.utils.text_postprocessors import first_capital_postprocess
ocnli_reader_cfg = dict( ocnli_reader_cfg = dict(
input_columns=["sentence1", "sentence2"], input_columns=["sentence1", "sentence2"],
...@@ -28,7 +29,7 @@ ocnli_infer_cfg = dict( ...@@ -28,7 +29,7 @@ ocnli_infer_cfg = dict(
ocnli_eval_cfg = dict( ocnli_eval_cfg = dict(
evaluator=dict(type=AccEvaluator), evaluator=dict(type=AccEvaluator),
pred_role="BOT", pred_role="BOT",
pred_postprocessor=dict(type="first-capital"), pred_postprocessor=dict(type=first_capital_postprocess),
) )
ocnli_datasets = [ ocnli_datasets = [
......
...@@ -3,6 +3,7 @@ from opencompass.openicl.icl_retriever import ZeroRetriever ...@@ -3,6 +3,7 @@ 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 AccEvaluator from opencompass.openicl.icl_evaluator import AccEvaluator
from opencompass.datasets import AFQMCDataset_V2 from opencompass.datasets import AFQMCDataset_V2
from opencompass.utils.text_postprocessors import first_capital_postprocess
bustm_reader_cfg = dict( bustm_reader_cfg = dict(
input_columns=["sentence1", "sentence2"], input_columns=["sentence1", "sentence2"],
...@@ -27,7 +28,7 @@ bustm_infer_cfg = dict( ...@@ -27,7 +28,7 @@ bustm_infer_cfg = dict(
bustm_eval_cfg = dict( bustm_eval_cfg = dict(
evaluator=dict(type=AccEvaluator), evaluator=dict(type=AccEvaluator),
pred_role="BOT", pred_role="BOT",
pred_postprocessor=dict(type="first-capital"), pred_postprocessor=dict(type=first_capital_postprocess),
) )
bustm_datasets = [ bustm_datasets = [
......
...@@ -3,6 +3,7 @@ from opencompass.openicl.icl_retriever import ZeroRetriever ...@@ -3,6 +3,7 @@ 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 AccEvaluator from opencompass.openicl.icl_evaluator import AccEvaluator
from opencompass.datasets import CHIDDataset_V2 from opencompass.datasets import CHIDDataset_V2
from opencompass.utils.text_postprocessors import first_capital_postprocess
chid_reader_cfg = dict( chid_reader_cfg = dict(
input_columns=["content","A","B","C","D","E","F","G"], input_columns=["content","A","B","C","D","E","F","G"],
...@@ -27,7 +28,7 @@ chid_infer_cfg = dict( ...@@ -27,7 +28,7 @@ chid_infer_cfg = dict(
chid_eval_cfg = dict( chid_eval_cfg = dict(
evaluator=dict(type=AccEvaluator), evaluator=dict(type=AccEvaluator),
pred_role="BOT", pred_role="BOT",
pred_postprocessor=dict(type="first-capital"), pred_postprocessor=dict(type=first_capital_postprocess),
) )
chid_datasets = [ chid_datasets = [
......
...@@ -3,6 +3,7 @@ from opencompass.openicl.icl_retriever import ZeroRetriever ...@@ -3,6 +3,7 @@ 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 AccEvaluator from opencompass.openicl.icl_evaluator import AccEvaluator
from opencompass.datasets import CluewscDataset_V2 from opencompass.datasets import CluewscDataset_V2
from opencompass.utils.text_postprocessors import first_capital_postprocess
cluewsc_reader_cfg = dict( cluewsc_reader_cfg = dict(
input_columns=["span1", "span2", "text", "new_text"], input_columns=["span1", "span2", "text", "new_text"],
...@@ -27,7 +28,7 @@ cluewsc_infer_cfg = dict( ...@@ -27,7 +28,7 @@ cluewsc_infer_cfg = dict(
cluewsc_eval_cfg = dict( cluewsc_eval_cfg = dict(
evaluator=dict(type=AccEvaluator), evaluator=dict(type=AccEvaluator),
pred_role="BOT", pred_role="BOT",
pred_postprocessor=dict(type="first-capital"), pred_postprocessor=dict(type=first_capital_postprocess),
) )
cluewsc_datasets = [ cluewsc_datasets = [
......
from mmengine.config import read_base from mmengine.config import read_base
with read_base(): with read_base():
from .FewCLUE_csl_gen_87f4a8 import csl_datasets # noqa: F401, F403 from .FewCLUE_csl_gen_28b223 import csl_datasets # noqa: F401, F403
from opencompass.openicl.icl_prompt_template import PromptTemplate
from opencompass.openicl.icl_retriever import ZeroRetriever
from opencompass.openicl.icl_inferencer import GenInferencer
from opencompass.openicl.icl_evaluator import AccEvaluator
from opencompass.datasets import CslDataset_V2
from opencompass.utils.text_postprocessors import first_capital_postprocess
csl_reader_cfg = dict(
input_columns=["abst", "keywords"],
output_column="label",
)
csl_infer_cfg = dict(
prompt_template=dict(
type=PromptTemplate,
template=dict(round=[
dict(
role="HUMAN",
prompt=
"摘要是对论文内容不加注释和评论的简短陈述,要求扼要地说明研究工作的目的、研究方法和最终结论等。\n关键词是一篇学术论文的核心词汇,一般由一系列名词组成。关键词在全文中应有较高出现频率,且能起到帮助文献检索的作用。\n摘要:{abst}\n关键词:{keywords}\n请问上述关键词是否匹配摘要且符合要求?\nA. 否\nB. 是\n请从”A“,”B“中进行选择。\n答:"
)
]),
),
retriever=dict(type=ZeroRetriever),
inferencer=dict(type=GenInferencer),
)
csl_eval_cfg = dict(
evaluator=dict(type=AccEvaluator),
pred_role="BOT",
pred_postprocessor=dict(type=first_capital_postprocess),
)
csl_datasets = [
dict(
abbr="csl_dev",
type=CslDataset_V2,
path="./data/FewCLUE/csl/dev_few_all.json",
reader_cfg=csl_reader_cfg,
infer_cfg=csl_infer_cfg,
eval_cfg=csl_eval_cfg,
),
dict(
abbr="csl_test",
type=CslDataset_V2,
path="./data/FewCLUE/csl/test_public.json",
reader_cfg=csl_reader_cfg,
infer_cfg=csl_infer_cfg,
eval_cfg=csl_eval_cfg,
),
]
...@@ -3,6 +3,7 @@ from opencompass.openicl.icl_retriever import ZeroRetriever ...@@ -3,6 +3,7 @@ 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 AccEvaluator from opencompass.openicl.icl_evaluator import AccEvaluator
from opencompass.datasets import CslDataset_V2 from opencompass.datasets import CslDataset_V2
from opencompass.utils.text_postprocessors import first_capital_postprocess
csl_reader_cfg = dict( csl_reader_cfg = dict(
input_columns=["abst", "keywords"], input_columns=["abst", "keywords"],
...@@ -27,7 +28,7 @@ csl_infer_cfg = dict( ...@@ -27,7 +28,7 @@ csl_infer_cfg = dict(
csl_eval_cfg = dict( csl_eval_cfg = dict(
evaluator=dict(type=AccEvaluator), evaluator=dict(type=AccEvaluator),
pred_role="BOT", pred_role="BOT",
pred_postprocessor=dict(type="first-capital"), pred_postprocessor=dict(type=first_capital_postprocess),
) )
csl_datasets = [ csl_datasets = [
......
...@@ -3,6 +3,7 @@ from opencompass.openicl.icl_retriever import ZeroRetriever ...@@ -3,6 +3,7 @@ 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 AccEvaluator from opencompass.openicl.icl_evaluator import AccEvaluator
from opencompass.datasets import eprstmtDataset_V2 from opencompass.datasets import eprstmtDataset_V2
from opencompass.utils.text_postprocessors import first_capital_postprocess
eprstmt_reader_cfg = dict( eprstmt_reader_cfg = dict(
input_columns=["sentence"], output_column="label", test_split="train") input_columns=["sentence"], output_column="label", test_split="train")
...@@ -25,7 +26,7 @@ eprstmt_infer_cfg = dict( ...@@ -25,7 +26,7 @@ eprstmt_infer_cfg = dict(
eprstmt_eval_cfg = dict( eprstmt_eval_cfg = dict(
evaluator=dict(type=AccEvaluator), evaluator=dict(type=AccEvaluator),
pred_role="BOT", pred_role="BOT",
pred_postprocessor=dict(type="first-capital"), pred_postprocessor=dict(type=first_capital_postprocess),
) )
eprstmt_datasets = [ eprstmt_datasets = [
......
...@@ -3,6 +3,7 @@ from opencompass.openicl.icl_retriever import ZeroRetriever ...@@ -3,6 +3,7 @@ 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 AccEvaluator from opencompass.openicl.icl_evaluator import AccEvaluator
from opencompass.datasets import cmnliDataset_V2 from opencompass.datasets import cmnliDataset_V2
from opencompass.utils.text_postprocessors import first_capital_postprocess
ocnli_fc_reader_cfg = dict( ocnli_fc_reader_cfg = dict(
input_columns=["sentence1", "sentence2"], input_columns=["sentence1", "sentence2"],
...@@ -26,7 +27,7 @@ ocnli_fc_infer_cfg = dict( ...@@ -26,7 +27,7 @@ ocnli_fc_infer_cfg = dict(
ocnli_fc_eval_cfg = dict( ocnli_fc_eval_cfg = dict(
evaluator=dict(type=AccEvaluator), evaluator=dict(type=AccEvaluator),
pred_role="BOT", pred_role="BOT",
pred_postprocessor=dict(type="first-capital"), pred_postprocessor=dict(type=first_capital_postprocess),
) )
ocnli_fc_datasets = [ ocnli_fc_datasets = [
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment