From 60fb13cddf75b7ae1e37d861ee20c3c7ffc0c5b0 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Thu, 18 Jul 2019 14:53:44 +0200 Subject: [PATCH] Added debug print for shape of true positives Signed-off-by: Jim Martens --- src/twomartens/masterthesis/evaluate.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/twomartens/masterthesis/evaluate.py b/src/twomartens/masterthesis/evaluate.py index de661d7..be47ed2 100644 --- a/src/twomartens/masterthesis/evaluate.py +++ b/src/twomartens/masterthesis/evaluate.py @@ -324,6 +324,7 @@ def get_precision_recall(number_gt_per_class: np.ndarray, diff_to_largest_class = cumulative_precision_micro.shape[0] - cumulative_precision.shape[0] if diff_to_largest_class: print(f"Diff to largest: {diff_to_largest_class}") + print(f"Shape: {tp.shape}") repeated_last_precision = np.tile(cumulative_precision[-1], diff_to_largest_class) repeated_last_recall = np.tile(cumulative_recall[-1], diff_to_largest_class) extended_precision = np.concatenate((cumulative_precision, repeated_last_precision))