remove compute_confusion_matrix

luxonis · Nov 27, 2024 · 7d4e469 · 7d4e469
1 parent 985233a
commit 7d4e469
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/luxonis_train/attached_modules/metrics/confusion_matrix.py b/luxonis_train/attached_modules/metrics/confusion_matrix.py
@@ -83,9 +83,6 @@ def __init__(
                 ),  # +1 for background
                 dist_reduce_fx="sum",
             )
-        self.compute_confusion_matrix = MulticlassConfusionMatrix(
-            num_classes=self.n_classes
-        )
 
     def prepare(
         self, inputs: Packet[Tensor], labels: Labels