Skip to content

GitLab

"examples/vscode:/vscode.git/clone" did not exist on "2b04ec2ff7270d2044410378b04d85a194fa3d4a"
Commit 8ec42a5c authored by WenmuZhou's avatar WenmuZhou
Browse files

move import seqeval to get_metric fun

parent cd7b2ea9
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
ppocr/metrics/vqa_token_ser_metric.py
View file @ 8ec42a5c
...@@ -18,7 +18,6 @@ from __future__ import print_function ...@@ -18,7 +18,6 @@ from __future__ import print_function
import numpy as np import numpy as np
import paddle import paddle
from seqeval.metrics import f1_score, precision_score, recall_score
__all__ = ['KIEMetric'] __all__ = ['KIEMetric']
...@@ -34,6 +33,7 @@ class VQASerTokenMetric(object): ...@@ -34,6 +33,7 @@ class VQASerTokenMetric(object):
self.gt_list.extend(labels) self.gt_list.extend(labels)
def get_metric(self): def get_metric(self):
from seqeval.metrics import f1_score, precision_score, recall_score
metircs = { metircs = {
"precision": precision_score(self.gt_list, self.pred_list), "precision": precision_score(self.gt_list, self.pred_list),
"recall": recall_score(self.gt_list, self.pred_list), "recall": recall_score(self.gt_list, self.pred_list),
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment