diff --git a/opencompass/datasets/gsm8k.py b/opencompass/datasets/gsm8k.py index ffe0fdd8a..a408ed6e9 100644 --- a/opencompass/datasets/gsm8k.py +++ b/opencompass/datasets/gsm8k.py @@ -43,10 +43,10 @@ def gsm8k_dataset_postprocess(text: str) -> str: @TEXT_POSTPROCESSORS.register_module('gsm8k') def gsm8k_postprocess(text: str) -> str: text = text.split('Question:')[0] - numbers = re.findall(r'\-?\d+\.\d+|\-?\d+', text) + numbers = re.findall(r'-?\d[\d,].?\d', text) if not numbers: - return 'NULL' - return numbers[-1] + return 'NULL' + return numbers[-1].replace(",","") class Gsm8kEvaluator(BaseEvaluator):