1 files changed, 8 insertions, 3 deletions
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()