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

Modified mean average precision to only use classes with ground truth 

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens 2019-07-15 12:44:52 +02:00
parent df2faaffe9
commit 5aac93f444
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -382,4 +382,6 @@ def get_mean_average_precision(average_precisions: List[float]) -> float:
Returns:
mean average precision over all classes
"""
return np.average(average_precisions[1:])
average_precisions = np.copy(average_precisions)
average_precisions = average_precisions[average_precisions != 0.0]
return np.average(average_precisions[:])