mirror of
https://github.com/open-compass/opencompass.git
synced 2025-05-30 16:03:24 +08:00
Update
This commit is contained in:
parent
580a2b7980
commit
8274603540
@ -204,7 +204,11 @@ def math_postprocess_v2(text: str) -> str:
|
||||
@ICL_EVALUATORS.register_module()
|
||||
class MATHEvaluator(BaseEvaluator):
|
||||
|
||||
def __init__(self, version='v1'):
|
||||
def __init__(self,
|
||||
version='v1',
|
||||
pred_postprocessor=None): # 可能需要接收父类__init__的参数
|
||||
super().__init__(
|
||||
pred_postprocessor=pred_postprocessor) # 调用父类的__init__
|
||||
assert version in ['v1', 'v2']
|
||||
self.version = version
|
||||
|
||||
|
@ -280,7 +280,11 @@ class MusrDataset(BaseDataset):
|
||||
@ICL_EVALUATORS.register_module()
|
||||
class MusrEvaluator(BaseEvaluator):
|
||||
|
||||
def __init__(self, answer_index_modifier=1, self_consistency_n=1):
|
||||
def __init__(self,
|
||||
answer_index_modifier=1,
|
||||
self_consistency_n=1,
|
||||
pred_postprocessor=None):
|
||||
super().__init__(pred_postprocessor=pred_postprocessor)
|
||||
self.answer_index_modifier = answer_index_modifier
|
||||
self.self_consistency_n = self_consistency_n
|
||||
|
||||
|
@ -93,7 +93,8 @@ class BaseEvaluator:
|
||||
return g_passk_details
|
||||
|
||||
def pred_postprocess(self, predictions: List) -> Dict:
|
||||
if self.pred_postprocessor is None:
|
||||
if not hasattr(
|
||||
self, 'pred_postprocessor') or self.pred_postprocessor is None:
|
||||
return predictions
|
||||
else:
|
||||
kwargs = deepcopy(self.pred_postprocessor)
|
||||
|
Loading…
Reference in New Issue
Block a user