From 19872e3f8e94444afe00ac1f372f46ec101f0812 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eldar=20Kurti=C4=87?= <8884008+eldarkurtic@users.noreply.github.com> Date: Thu, 8 Jan 2026 23:07:13 +0100 Subject: [PATCH] Fix precision and F1 score calculations --- guardbench/evaluate.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/guardbench/evaluate.py b/guardbench/evaluate.py index fd986cb..fac1c24 100644 --- a/guardbench/evaluate.py +++ b/guardbench/evaluate.py @@ -62,9 +62,9 @@ def evaluate(y_true: dict, y_pred_prob: dict, threshold: float = 0.5) -> dict: mcc = matthews_corrcoef(y_true, y_pred_binary) return { - "precision": float(precision) if n > 0 else 0.0, + "precision": float(precision), "recall": float(recall), - "f1": float(f1) if n > 0 else 0.0, + "f1": float(f1), "mcc": float(mcc) if n > 0 else 0.0, "auprc": float(auprc) if n > 0 else 0.0, "sensitivity": float(sensitivity),