diff --git a/results/roc.py b/results/roc.py index 4b03993a26bdd3c7111598393a81e18682438945..21ea3789bb5fbc76e4e42a9e7dd51b236ded825e 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)}}}