➕Add reset methods to metrics (#488)

openvinotoolkit · Aug 11, 2022 · a0e040d · a0e040d
1 parent 32ebbc3
commit a0e040d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/anomalib/utils/metrics/adaptive_threshold.py b/anomalib/utils/metrics/adaptive_threshold.py
@@ -48,3 +48,7 @@ def compute(self) -> torch.Tensor:
         else:
             self.value = thresholds[torch.argmax(f1_score)]
         return self.value
+
+    def reset(self) -> None:
+        """Reset the metric."""
+        self.precision_recall_curve.reset()
diff --git a/anomalib/utils/metrics/optimal_f1.py b/anomalib/utils/metrics/optimal_f1.py
@@ -44,3 +44,7 @@ def compute(self) -> torch.Tensor:
         self.threshold = thresholds[torch.argmax(f1_score)]
         optimal_f1_score = torch.max(f1_score)
         return optimal_f1_score
+
+    def reset(self) -> None:
+        """Reset the metric."""
+        self.precision_recall_curve.reset()