From 073d3319c8625e9aecb4398c07c0ad3ddcaf753a Mon Sep 17 00:00:00 2001 From: wangxiao1021 Date: Thu, 20 Feb 2020 14:59:09 +0800 Subject: [PATCH] update evaluate-intent.py --- examples/multi-task/evaluate-intent.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/multi-task/evaluate-intent.py b/examples/multi-task/evaluate-intent.py index 7c491cd..5d2feb5 100644 --- a/examples/multi-task/evaluate-intent.py +++ b/examples/multi-task/evaluate-intent.py @@ -15,7 +15,7 @@ def f1(preds, labels): tn = np.sum((labels == '0') & (preds == '0')) fp = np.sum((labels == '0') & (preds == '1')) fn = np.sum((labels == '1') & (preds == '0')) - p = tp * 1.0 / (tp + fp) + p = tp * 1.0 / (tp + fp) * 1.0 r = tp * 1.0 / (tp + fn) * 1.0 f1 = (2 * p * r) / (p + r + 1e-8) return f1 -- GitLab