From d8930a4e3aa2aa4421a7fa04e8e2086206dc6752 Mon Sep 17 00:00:00 2001
From: Pierre LOTTE <pierrelotte.dev@gmail.com>
Date: Mon, 30 Sep 2024 08:59:24 +0200
Subject: [PATCH] Fix bug in results computation
---
results/roc.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/results/roc.py b/results/roc.py
index 4b03993..21ea378 100644
--- a/results/roc.py
+++ b/results/roc.py
@@ -37,7 +37,7 @@ class ROCResults(BaseResults):
y_pred_path = f"{self.path}/{config_name}/results_{algo}/anomaly_scores_dataset_{algo}.ts"
roc, f1 = self.__compute_score(labels, y_pred_path)
- if algo in self.result["roc"]:
+ if algo in self.result:
self.result[algo][config_name] = {"classic": {"roc": round(roc, 4), "f1": round(f1, 4)}}
else:
self.result[algo] = {config_name: {"classic": {"roc": round(roc, 4), "f1": round(f1, 4)}}}
--
GitLab