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

Fixed guardian if to work with numpy arrays

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-06-10 16:55:07 +02:00
parent 488703bb26
commit ca90ea9f17
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -293,7 +293,7 @@ def get_f1_score(cumulative_precisions: List[np.ndarray],
for class_id in range(1, nr_classes + 1):
cumulative_precision = cumulative_precisions[class_id]
cumulative_recall = cumulative_recalls[class_id]
if (cumulative_precision + cumulative_recall) == 0:
if not np.count_nonzero(cumulative_precision + cumulative_recall):
cumulative_f1_scores.append([])
continue
f1_score = 2 * ((cumulative_precision * cumulative_recall) / (cumulative_precision + cumulative_recall))