[Enhancement] Remove reduntant code snippet in F1Metric (#1278)

pull/1281/head
Tong Gao 2022-08-11 18:00:15 +08:00 committed by GitHub
parent 7b25b62c21
commit 792cb26924
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 3 deletions

View File

@ -62,6 +62,7 @@ class F1Metric(BaseMetric):
assert isinstance(mode, (list, str))
assert not (len(cared_classes) > 0 and len(ignored_classes) > 0), \
'cared_classes and ignored_classes cannot be both non-empty'
if isinstance(mode, str):
mode = [mode]
assert set(mode).issubset({'micro', 'macro'})
@ -82,9 +83,6 @@ class F1Metric(BaseMetric):
self.cared_labels = list(range(num_classes))
self.num_classes = num_classes
self.key = key
if isinstance(mode, str):
mode = [mode]
self.mode = mode
def process(self, data_batch: Sequence[Dict],
predictions: Sequence[Dict]) -> None: