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

Modified mean average precision to only use classes with ground truth

Browse Source 
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
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: Returns:
mean average precision over all classes 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[:])