diff --git a/proeval/evaluator/predictor.py b/proeval/evaluator/predictor.py index 87302f9..30d1b7d 100644 --- a/proeval/evaluator/predictor.py +++ b/proeval/evaluator/predictor.py @@ -197,10 +197,10 @@ def _cmp(pred, gt): return 0.0 if str(pred).strip() == str(gt).strip() else 1.0 def _gt(ans): - m = re.search(r"####\s*([0-9,]+(?:\.[0-9]+)?)", str(ans)) + m = re.search(r"####\s*([+-]?[0-9,]+(?:\.[0-9]+)?)", str(ans)) if m: return m.group(1).replace(",", "") - m = re.search(r"([0-9,]+(?:\.[0-9]+)?)", str(ans)) + m = re.search(r"([+-]?[0-9,]+(?:\.[0-9]+)?)", str(ans)) return m.group(1).replace(",", "") if m else str(ans).strip() return DatasetConfig(