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()
|
@ICL_EVALUATORS.register_module()
|
||||||
class MATHEvaluator(BaseEvaluator):
|
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']
|
assert version in ['v1', 'v2']
|
||||||
self.version = version
|
self.version = version
|
||||||
|
|
||||||
|
@ -280,7 +280,11 @@ class MusrDataset(BaseDataset):
|
|||||||
@ICL_EVALUATORS.register_module()
|
@ICL_EVALUATORS.register_module()
|
||||||
class MusrEvaluator(BaseEvaluator):
|
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.answer_index_modifier = answer_index_modifier
|
||||||
self.self_consistency_n = self_consistency_n
|
self.self_consistency_n = self_consistency_n
|
||||||
|
|
||||||
|
@ -93,7 +93,8 @@ class BaseEvaluator:
|
|||||||
return g_passk_details
|
return g_passk_details
|
||||||
|
|
||||||
def pred_postprocess(self, predictions: List) -> Dict:
|
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
|
return predictions
|
||||||
else:
|
else:
|
||||||
kwargs = deepcopy(self.pred_postprocessor)
|
kwargs = deepcopy(self.pred_postprocessor)
|
||||||
|
Loading…
Reference in New Issue
Block a user