mirror of
https://github.com/open-compass/opencompass.git
synced 2025-05-30 16:03:24 +08:00
29 lines
733 B
Python
29 lines
733 B
Python
from opencompass.registry import TEXT_POSTPROCESSORS
|
|
|
|
|
|
@TEXT_POSTPROCESSORS.register_module('gsm8k_dataset')
|
|
def gsm8k_dataset_postprocess(text: str) -> str:
|
|
return text.split('#### ')[1].replace(',', '')
|
|
|
|
|
|
@TEXT_POSTPROCESSORS.register_module('gsm8k')
|
|
def gsm8k_postprocess(text: str) -> str:
|
|
text = text.split('\n\n')[0]
|
|
text = text.split(' ')[::-1]
|
|
flag = False
|
|
ret = ''
|
|
for i in range(len(text)):
|
|
s = text[i]
|
|
for i in range(len(s)):
|
|
if s[i].isdigit():
|
|
flag = True
|
|
ret = s
|
|
break
|
|
if flag:
|
|
break
|
|
ret1 = ''
|
|
for i in range(len(ret)):
|
|
if ret[i].isdigit():
|
|
ret1 += ret[i]
|
|
return ret1
|