From 7d4e469633274014e430e5b27c3489f131fe2156 Mon Sep 17 00:00:00 2001 From: Jernej Sabadin Date: Wed, 27 Nov 2024 14:30:18 +0100 Subject: [PATCH] remove compute_confusion_matrix --- luxonis_train/attached_modules/metrics/confusion_matrix.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/luxonis_train/attached_modules/metrics/confusion_matrix.py b/luxonis_train/attached_modules/metrics/confusion_matrix.py index 3d9ddd18..e88e27d8 100644 --- 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