From 76d032e53e4e84c1c0036daaadeab2821be0730a Mon Sep 17 00:00:00 2001
From: adamjanovsky
Date: Fri, 17 Feb 2023 14:11:57 +0100
Subject: more work on PoC reference annotator

---
 src/sec_certs/utils/nlp.py | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

(limited to 'src/sec_certs/utils')

diff --git a/src/sec_certs/utils/nlp.py b/src/sec_certs/utils/nlp.py
index 67ae2869..496b71b8 100644
--- a/src/sec_certs/utils/nlp.py
+++ b/src/sec_certs/utils/nlp.py
@@ -1,8 +1,13 @@
+import numpy as np
 from sklearn.metrics import precision_score, recall_score
 
 
-def prec_recall_metric(y_pred, y_test):
+def prec_recall_metric(y_pred, y_true):
     return {
-        "precision": precision_score(y_test, y_pred, zero_division="warn", average="micro"),
-        "recall": recall_score(y_test, y_pred, zero_division="warn", average="micro"),
+        "precision": precision_score(y_true, y_pred, zero_division="warn", average="micro"),
+        "recall": recall_score(y_true, y_pred, zero_division="warn", average="micro"),
     }
+
+
+def softmax(x):
+    return np.exp(x - np.max(x)) / np.exp(x - np.max(x)).sum()
-- 
cgit v1.3.1