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

Change f1 score to single value

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-15 14:46:09 +02:00
parent c56440cb49
commit 590b5e95ca
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -305,7 +305,8 @@ def get_f1_score(cumulative_precisions: List[np.ndarray],
if not np.count_nonzero(cumulative_precision + cumulative_recall): if not np.count_nonzero(cumulative_precision + cumulative_recall):
cumulative_f1_scores.append([]) cumulative_f1_scores.append([])
continue continue
f1_score = 2 * ((cumulative_precision * cumulative_recall) / (cumulative_precision + cumulative_recall + 0.001)) f1_score = 2 * ((cumulative_precision[-1] * cumulative_recall[-1]) /
(cumulative_precision[-1] + cumulative_recall[-1] + 0.001))
cumulative_f1_scores.append(f1_score) cumulative_f1_scores.append(f1_score)
return cumulative_f1_scores return cumulative_f1_scores