From d74bb2ac7f3b42ec611cc171ec0c474bc4f36cd3 Mon Sep 17 00:00:00 2001 From: MaiziXiao Date: Fri, 18 Apr 2025 07:27:13 +0000 Subject: [PATCH] update --- dataset-index.yml | 2 +- .../datasets/ChemBench/ChemBench_gen.py | 79 +------------ .../ChemBench/ChemBench_gen_a9f753.py | 77 ++++++++++++ .../ChemBench/ChemBench_llmjudge_gen.py | 110 +----------------- .../ChemBench_llmjudge_gen_c584cf.py | 108 +++++++++++++++++ ..._gen.py => smolinstruct_fts_gen_5774b5.py} | 0 .../datasets/SmolInstruct/smolinstruct_gen.py | 8 +- .../SmolInstruct/smolinstruct_gen_c84c18.py | 10 ++ ...n.py => smolinstruct_meteor_gen_065150.py} | 0 ...c_gen.py => smolinstruct_nc_gen_c84c18.py} | 0 ...n.py => smolinstruct_pp_acc_gen_8607a3.py} | 0 ...gen.py => smolinstruct_rmse_gen_0fcc6b.py} | 0 12 files changed, 203 insertions(+), 191 deletions(-) create mode 100644 opencompass/configs/datasets/ChemBench/ChemBench_gen_a9f753.py create mode 100644 opencompass/configs/datasets/ChemBench/ChemBench_llmjudge_gen_c584cf.py rename opencompass/configs/datasets/SmolInstruct/{smolinstruct_fts_gen.py => smolinstruct_fts_gen_5774b5.py} (100%) create mode 100644 opencompass/configs/datasets/SmolInstruct/smolinstruct_gen_c84c18.py rename opencompass/configs/datasets/SmolInstruct/{smolinstruct_meteor_gen.py => smolinstruct_meteor_gen_065150.py} (100%) rename opencompass/configs/datasets/SmolInstruct/{smolinstruct_nc_gen.py => smolinstruct_nc_gen_c84c18.py} (100%) rename opencompass/configs/datasets/SmolInstruct/{smolinstruct_pp_acc_gen.py => smolinstruct_pp_acc_gen_8607a3.py} (100%) rename opencompass/configs/datasets/SmolInstruct/{smolinstruct_rmse_gen.py => smolinstruct_rmse_gen_0fcc6b.py} (100%) diff --git a/dataset-index.yml b/dataset-index.yml index 5b4e3b28..6e2c3fc9 100644 --- a/dataset-index.yml +++ b/dataset-index.yml @@ -1015,5 +1015,5 @@ name: SmolInstruct category: Science /Chemistry paper: https://arxiv.org/pdf/2402.09391 - configpath: 'opencompass/configs/datasets/SmolInstruct/smolinstruct_gen.py' + configpath: opencompass/configs/datasets/SmolInstruct/smolinstruct_gen.py configpath_llmjudge: '' diff --git a/opencompass/configs/datasets/ChemBench/ChemBench_gen.py b/opencompass/configs/datasets/ChemBench/ChemBench_gen.py index efc9df48..66923794 100644 --- a/opencompass/configs/datasets/ChemBench/ChemBench_gen.py +++ b/opencompass/configs/datasets/ChemBench/ChemBench_gen.py @@ -1,77 +1,4 @@ -from opencompass.openicl.icl_prompt_template import PromptTemplate -from opencompass.openicl.icl_retriever import FixKRetriever -from opencompass.openicl.icl_inferencer import GenInferencer -from opencompass.openicl.icl_evaluator import AccEvaluator -from opencompass.datasets import ChemBenchDataset -from opencompass.utils.text_postprocessors import first_capital_postprocess +from mmengine.config import read_base - -chembench_reader_cfg = dict( - input_columns=['input', 'A', 'B', 'C', 'D'], - output_column='target', - train_split='dev') - -chembench_all_sets = [ - 'Name_Conversion', - 'Property_Prediction', - 'Mol2caption', - 'Caption2mol', - 'Product_Prediction', - 'Retrosynthesis', - 'Yield_Prediction', - 'Temperature_Prediction', - 'Solvent_Prediction' -] - - -chembench_datasets = [] -for _name in chembench_all_sets: - # _hint = f'There is a single choice question about {_name.replace("_", " ")}. Answer the question by replying A, B, C or D.' - _hint = f'There is a single choice question about chemistry. Answer the question by replying A, B, C or D.' - - chembench_infer_cfg = dict( - ice_template=dict( - type=PromptTemplate, - template=dict(round=[ - dict( - role='HUMAN', - prompt= - f'{_hint}\nQuestion: {{input}}\nA. {{A}}\nB. {{B}}\nC. {{C}}\nD. {{D}}\nAnswer: ' - ), - dict(role='BOT', prompt='{target}\n') - ]), - ), - prompt_template=dict( - type=PromptTemplate, - template=dict( - begin='', - round=[ - dict( - role='HUMAN', - prompt= - f'{_hint}\nQuestion: {{input}}\nA. {{A}}\nB. {{B}}\nC. {{C}}\nD. {{D}}\nAnswer: ' - ), - ], - ), - ice_token='', - ), - retriever=dict(type=FixKRetriever, fix_id_list=[0, 1, 2, 3, 4]), - inferencer=dict(type=GenInferencer), - ) - - chembench_eval_cfg = dict( - evaluator=dict(type=AccEvaluator), - pred_postprocessor=dict(type=first_capital_postprocess)) - - chembench_datasets.append( - dict( - abbr=f'ChemBench_{_name}', - type=ChemBenchDataset, - path='opencompass/ChemBench4K', - name=_name, - reader_cfg=chembench_reader_cfg, - infer_cfg=chembench_infer_cfg, - eval_cfg=chembench_eval_cfg, - )) - -del _name, _hint +with read_base(): + from .ChemBench_gen_a9f753 import chembench_datasets # noqa: F401, F403 \ No newline at end of file diff --git a/opencompass/configs/datasets/ChemBench/ChemBench_gen_a9f753.py b/opencompass/configs/datasets/ChemBench/ChemBench_gen_a9f753.py new file mode 100644 index 00000000..efc9df48 --- /dev/null +++ b/opencompass/configs/datasets/ChemBench/ChemBench_gen_a9f753.py @@ -0,0 +1,77 @@ +from opencompass.openicl.icl_prompt_template import PromptTemplate +from opencompass.openicl.icl_retriever import FixKRetriever +from opencompass.openicl.icl_inferencer import GenInferencer +from opencompass.openicl.icl_evaluator import AccEvaluator +from opencompass.datasets import ChemBenchDataset +from opencompass.utils.text_postprocessors import first_capital_postprocess + + +chembench_reader_cfg = dict( + input_columns=['input', 'A', 'B', 'C', 'D'], + output_column='target', + train_split='dev') + +chembench_all_sets = [ + 'Name_Conversion', + 'Property_Prediction', + 'Mol2caption', + 'Caption2mol', + 'Product_Prediction', + 'Retrosynthesis', + 'Yield_Prediction', + 'Temperature_Prediction', + 'Solvent_Prediction' +] + + +chembench_datasets = [] +for _name in chembench_all_sets: + # _hint = f'There is a single choice question about {_name.replace("_", " ")}. Answer the question by replying A, B, C or D.' + _hint = f'There is a single choice question about chemistry. Answer the question by replying A, B, C or D.' + + chembench_infer_cfg = dict( + ice_template=dict( + type=PromptTemplate, + template=dict(round=[ + dict( + role='HUMAN', + prompt= + f'{_hint}\nQuestion: {{input}}\nA. {{A}}\nB. {{B}}\nC. {{C}}\nD. {{D}}\nAnswer: ' + ), + dict(role='BOT', prompt='{target}\n') + ]), + ), + prompt_template=dict( + type=PromptTemplate, + template=dict( + begin='', + round=[ + dict( + role='HUMAN', + prompt= + f'{_hint}\nQuestion: {{input}}\nA. {{A}}\nB. {{B}}\nC. {{C}}\nD. {{D}}\nAnswer: ' + ), + ], + ), + ice_token='', + ), + retriever=dict(type=FixKRetriever, fix_id_list=[0, 1, 2, 3, 4]), + inferencer=dict(type=GenInferencer), + ) + + chembench_eval_cfg = dict( + evaluator=dict(type=AccEvaluator), + pred_postprocessor=dict(type=first_capital_postprocess)) + + chembench_datasets.append( + dict( + abbr=f'ChemBench_{_name}', + type=ChemBenchDataset, + path='opencompass/ChemBench4K', + name=_name, + reader_cfg=chembench_reader_cfg, + infer_cfg=chembench_infer_cfg, + eval_cfg=chembench_eval_cfg, + )) + +del _name, _hint diff --git a/opencompass/configs/datasets/ChemBench/ChemBench_llmjudge_gen.py b/opencompass/configs/datasets/ChemBench/ChemBench_llmjudge_gen.py index d6fc7e46..969cbf04 100644 --- a/opencompass/configs/datasets/ChemBench/ChemBench_llmjudge_gen.py +++ b/opencompass/configs/datasets/ChemBench/ChemBench_llmjudge_gen.py @@ -1,108 +1,4 @@ -from opencompass.datasets.math import MATHDataset -from opencompass.openicl.icl_prompt_template import PromptTemplate -from opencompass.openicl.icl_retriever import ZeroRetriever -from opencompass.openicl.icl_inferencer import GenInferencer -from opencompass.evaluator import GenericLLMEvaluator -from opencompass.datasets import generic_llmjudge_postprocess -from opencompass.datasets import ChemBenchDataset +from mmengine.config import read_base - -chembench_reader_cfg = dict( - input_columns=['input', 'A', 'B', 'C', 'D'], - output_column='target', - train_split='dev') - - - -GRADER_TEMPLATE = """ - Please as a grading expert, judge whether the final answers given by the candidates below are consistent with the standard answers, that is, whether the candidates answered correctly. - - Here are some evaluation criteria: - 1. Please refer to the given standard answer. You don't need to re-generate the answer to the question because the standard answer has been given. You only need to judge whether the candidate's answer is consistent with the standard answer according to the form of the question. Don't try to answer the original question. You can assume that the standard answer is definitely correct. - 2. Because the candidate's answer may be different from the standard answer in the form of expression, before making a judgment, please understand the question and the standard answer first, and then judge whether the candidate's answer is correct, but be careful not to try to answer the original question. - 3. Some answers may contain multiple items, such as multiple-choice questions, multiple-select questions, fill-in-the-blank questions, etc. As long as the answer is the same as the standard answer, it is enough. For multiple-select questions and multiple-blank fill-in-the-blank questions, the candidate needs to answer all the corresponding options or blanks correctly to be considered correct. - 4. Some answers may be expressed in different ways, such as some answers may be a mathematical expression, some answers may be a textual description, as long as the meaning expressed is the same. And some formulas are expressed in different ways, but they are equivalent and correct. - 5. If the prediction is given with \\boxed{}, please ignore the \\boxed{} and only judge whether the candidate's answer is consistent with the standard answer. - - Please judge whether the following answers are consistent with the standard answer based on the above criteria. Grade the predicted answer of this new question as one of: - A: CORRECT - B: INCORRECT - Just return the letters "A" or "B", with no text around it. - - Here is your task. Simply reply with either CORRECT, INCORRECT. Don't apologize or correct yourself if there was a mistake; we are just trying to grade the answer. - - - : \n {{input}}\nA. {{A}}\nB. {{B}}\nC. {{C}}\nD. {{D}}\n\n\n - : \n{target}\n\n\n - : \n{prediction}\n\n\n - - Judging the correctness of candidates' answers: -""".strip() - -chembench_all_sets = [ - 'Name_Conversion', - 'Property_Prediction', - 'Mol2caption', - 'Caption2mol', - 'Product_Prediction', - 'Retrosynthesis', - 'Yield_Prediction', - 'Temperature_Prediction', - 'Solvent_Prediction' -] -_hint = f'There is a single choice question about chemistry. Answer the question by replying A, B, C or D.' - -chembench_datasets = [] -for _name in chembench_all_sets: - chembench_infer_cfg = dict( - prompt_template=dict( - type=PromptTemplate, - template=dict(round=[ - dict(role='HUMAN', prompt=f'{_hint}\nQuestion: {{input}}\nA. {{A}}\nB. {{B}}\nC. {{C}}\nD. {{D}}\nAnswer: ') - ])), - retriever=dict(type=ZeroRetriever), - inferencer=dict(type=GenInferencer) - ) - - # Evaluation configuration - chembench_eval_cfg = dict( - evaluator=dict( - type=GenericLLMEvaluator, - prompt_template=dict( - type=PromptTemplate, - template=dict( - begin=[ - dict( - role='SYSTEM', - fallback_role='HUMAN', - prompt="You are a helpful assistant who evaluates the correctness and quality of models' outputs.") - ], - round=[ - dict( - role='HUMAN', - prompt = GRADER_TEMPLATE - ), - ]), - ), - dataset_cfg=dict( - type=ChemBenchDataset, - path='/fs-computility/llm/xiaolinchen/opencompass_fork/data/ChemBench4K', - name=_name, - reader_cfg=chembench_reader_cfg, - ), - judge_cfg=dict(), - dict_postprocessor=dict(type=generic_llmjudge_postprocess), - ), - pred_role='BOT', - ) - - chembench_datasets.append( - dict( - abbr=f'ChemBench_{_name}', - type=ChemBenchDataset, - path='opencompass/ChemBench4K', - name=_name, - reader_cfg=chembench_reader_cfg, - infer_cfg=chembench_infer_cfg, - eval_cfg=chembench_eval_cfg, - )) \ No newline at end of file +with read_base(): + from .ChemBench_llmjudge_gen_c584cf import chembench_datasets # noqa: F401, F403 \ No newline at end of file diff --git a/opencompass/configs/datasets/ChemBench/ChemBench_llmjudge_gen_c584cf.py b/opencompass/configs/datasets/ChemBench/ChemBench_llmjudge_gen_c584cf.py new file mode 100644 index 00000000..d6fc7e46 --- /dev/null +++ b/opencompass/configs/datasets/ChemBench/ChemBench_llmjudge_gen_c584cf.py @@ -0,0 +1,108 @@ +from opencompass.datasets.math import MATHDataset +from opencompass.openicl.icl_prompt_template import PromptTemplate +from opencompass.openicl.icl_retriever import ZeroRetriever +from opencompass.openicl.icl_inferencer import GenInferencer +from opencompass.evaluator import GenericLLMEvaluator +from opencompass.datasets import generic_llmjudge_postprocess +from opencompass.datasets import ChemBenchDataset + + +chembench_reader_cfg = dict( + input_columns=['input', 'A', 'B', 'C', 'D'], + output_column='target', + train_split='dev') + + + +GRADER_TEMPLATE = """ + Please as a grading expert, judge whether the final answers given by the candidates below are consistent with the standard answers, that is, whether the candidates answered correctly. + + Here are some evaluation criteria: + 1. Please refer to the given standard answer. You don't need to re-generate the answer to the question because the standard answer has been given. You only need to judge whether the candidate's answer is consistent with the standard answer according to the form of the question. Don't try to answer the original question. You can assume that the standard answer is definitely correct. + 2. Because the candidate's answer may be different from the standard answer in the form of expression, before making a judgment, please understand the question and the standard answer first, and then judge whether the candidate's answer is correct, but be careful not to try to answer the original question. + 3. Some answers may contain multiple items, such as multiple-choice questions, multiple-select questions, fill-in-the-blank questions, etc. As long as the answer is the same as the standard answer, it is enough. For multiple-select questions and multiple-blank fill-in-the-blank questions, the candidate needs to answer all the corresponding options or blanks correctly to be considered correct. + 4. Some answers may be expressed in different ways, such as some answers may be a mathematical expression, some answers may be a textual description, as long as the meaning expressed is the same. And some formulas are expressed in different ways, but they are equivalent and correct. + 5. If the prediction is given with \\boxed{}, please ignore the \\boxed{} and only judge whether the candidate's answer is consistent with the standard answer. + + Please judge whether the following answers are consistent with the standard answer based on the above criteria. Grade the predicted answer of this new question as one of: + A: CORRECT + B: INCORRECT + Just return the letters "A" or "B", with no text around it. + + Here is your task. Simply reply with either CORRECT, INCORRECT. Don't apologize or correct yourself if there was a mistake; we are just trying to grade the answer. + + + : \n {{input}}\nA. {{A}}\nB. {{B}}\nC. {{C}}\nD. {{D}}\n\n\n + : \n{target}\n\n\n + : \n{prediction}\n\n\n + + Judging the correctness of candidates' answers: +""".strip() + +chembench_all_sets = [ + 'Name_Conversion', + 'Property_Prediction', + 'Mol2caption', + 'Caption2mol', + 'Product_Prediction', + 'Retrosynthesis', + 'Yield_Prediction', + 'Temperature_Prediction', + 'Solvent_Prediction' +] +_hint = f'There is a single choice question about chemistry. Answer the question by replying A, B, C or D.' + +chembench_datasets = [] +for _name in chembench_all_sets: + chembench_infer_cfg = dict( + prompt_template=dict( + type=PromptTemplate, + template=dict(round=[ + dict(role='HUMAN', prompt=f'{_hint}\nQuestion: {{input}}\nA. {{A}}\nB. {{B}}\nC. {{C}}\nD. {{D}}\nAnswer: ') + ])), + retriever=dict(type=ZeroRetriever), + inferencer=dict(type=GenInferencer) + ) + + # Evaluation configuration + chembench_eval_cfg = dict( + evaluator=dict( + type=GenericLLMEvaluator, + prompt_template=dict( + type=PromptTemplate, + template=dict( + begin=[ + dict( + role='SYSTEM', + fallback_role='HUMAN', + prompt="You are a helpful assistant who evaluates the correctness and quality of models' outputs.") + ], + round=[ + dict( + role='HUMAN', + prompt = GRADER_TEMPLATE + ), + ]), + ), + dataset_cfg=dict( + type=ChemBenchDataset, + path='/fs-computility/llm/xiaolinchen/opencompass_fork/data/ChemBench4K', + name=_name, + reader_cfg=chembench_reader_cfg, + ), + judge_cfg=dict(), + dict_postprocessor=dict(type=generic_llmjudge_postprocess), + ), + pred_role='BOT', + ) + + chembench_datasets.append( + dict( + abbr=f'ChemBench_{_name}', + type=ChemBenchDataset, + path='opencompass/ChemBench4K', + name=_name, + reader_cfg=chembench_reader_cfg, + infer_cfg=chembench_infer_cfg, + eval_cfg=chembench_eval_cfg, + )) \ No newline at end of file diff --git a/opencompass/configs/datasets/SmolInstruct/smolinstruct_fts_gen.py b/opencompass/configs/datasets/SmolInstruct/smolinstruct_fts_gen_5774b5.py similarity index 100% rename from opencompass/configs/datasets/SmolInstruct/smolinstruct_fts_gen.py rename to opencompass/configs/datasets/SmolInstruct/smolinstruct_fts_gen_5774b5.py diff --git a/opencompass/configs/datasets/SmolInstruct/smolinstruct_gen.py b/opencompass/configs/datasets/SmolInstruct/smolinstruct_gen.py index 3bd53d79..357062d5 100644 --- a/opencompass/configs/datasets/SmolInstruct/smolinstruct_gen.py +++ b/opencompass/configs/datasets/SmolInstruct/smolinstruct_gen.py @@ -1,10 +1,4 @@ from mmengine.config import read_base with read_base(): - from opencompass.configs.datasets.SmolInstruct.smolinstruct_nc_gen import nc_datasets - from opencompass.configs.datasets.SmolInstruct.smolinstruct_pp_acc_gen import pp_acc_datasets - from opencompass.configs.datasets.SmolInstruct.smolinstruct_rmse_gen import pp_rmse_datasets - from opencompass.configs.datasets.SmolInstruct.smolinstruct_fts_gen import fts_datasets - from opencompass.configs.datasets.SmolInstruct.smolinstruct_meteor_gen import meteor_datasets - -smolinstruct_datasets = nc_datasets + pp_rmse_datasets + pp_acc_datasets + meteor_datasets + fts_datasets + from smolinstruct_gen_c84c18 import smolinstruct_datasets diff --git a/opencompass/configs/datasets/SmolInstruct/smolinstruct_gen_c84c18.py b/opencompass/configs/datasets/SmolInstruct/smolinstruct_gen_c84c18.py new file mode 100644 index 00000000..8b02c1e6 --- /dev/null +++ b/opencompass/configs/datasets/SmolInstruct/smolinstruct_gen_c84c18.py @@ -0,0 +1,10 @@ +from mmengine.config import read_base + +with read_base(): + from opencompass.configs.datasets.SmolInstruct.smolinstruct_nc_gen_c84c18 import nc_datasets + from opencompass.configs.datasets.SmolInstruct.smolinstruct_pp_acc_gen_8607a3 import pp_acc_datasets + from opencompass.configs.datasets.SmolInstruct.smolinstruct_rmse_gen_0fcc6b import pp_rmse_datasets + from opencompass.configs.datasets.SmolInstruct.smolinstruct_fts_gen_5774b5 import fts_datasets + from opencompass.configs.datasets.SmolInstruct.smolinstruct_meteor_gen_065150 import meteor_datasets + +smolinstruct_datasets = nc_datasets + pp_rmse_datasets + pp_acc_datasets + meteor_datasets + fts_datasets diff --git a/opencompass/configs/datasets/SmolInstruct/smolinstruct_meteor_gen.py b/opencompass/configs/datasets/SmolInstruct/smolinstruct_meteor_gen_065150.py similarity index 100% rename from opencompass/configs/datasets/SmolInstruct/smolinstruct_meteor_gen.py rename to opencompass/configs/datasets/SmolInstruct/smolinstruct_meteor_gen_065150.py diff --git a/opencompass/configs/datasets/SmolInstruct/smolinstruct_nc_gen.py b/opencompass/configs/datasets/SmolInstruct/smolinstruct_nc_gen_c84c18.py similarity index 100% rename from opencompass/configs/datasets/SmolInstruct/smolinstruct_nc_gen.py rename to opencompass/configs/datasets/SmolInstruct/smolinstruct_nc_gen_c84c18.py diff --git a/opencompass/configs/datasets/SmolInstruct/smolinstruct_pp_acc_gen.py b/opencompass/configs/datasets/SmolInstruct/smolinstruct_pp_acc_gen_8607a3.py similarity index 100% rename from opencompass/configs/datasets/SmolInstruct/smolinstruct_pp_acc_gen.py rename to opencompass/configs/datasets/SmolInstruct/smolinstruct_pp_acc_gen_8607a3.py diff --git a/opencompass/configs/datasets/SmolInstruct/smolinstruct_rmse_gen.py b/opencompass/configs/datasets/SmolInstruct/smolinstruct_rmse_gen_0fcc6b.py similarity index 100% rename from opencompass/configs/datasets/SmolInstruct/smolinstruct_rmse_gen.py rename to opencompass/configs/datasets/SmolInstruct/smolinstruct_rmse_gen_0fcc6b.py