2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Added guardian function to catch case with zero ground truth

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-09 13:45:01 +02:00
parent c6e7dd3c50
commit bdd0862d68
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/twomartens/masterthesis/evaluate.py
View File

@ -263,6 +263,11 @@ def get_precision_recall(number_gt_per_class: np.ndarray,
# Iterate over all classes. # Iterate over all classes.
for class_id in range(1, nr_classes + 1): for class_id in range(1, nr_classes + 1):
if number_gt_per_class[class_id] == 0:
cumulative_precisions.append([])
cumulative_recalls.append([])
continue
tp = cumulative_true_positives[class_id] tp = cumulative_true_positives[class_id]
fp = cumulative_false_positives[class_id] fp = cumulative_false_positives[class_id]