Including all recall points.

rafaelpadilla · Jan 31, 2019 · 6c7c54d · 6c7c54d
1 parent aeb7217
commit 6c7c54d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/Evaluator.py b/lib/Evaluator.py
@@ -329,7 +329,7 @@ def ElevenPointInterpolatedAP(rec, prec):
         # For each recallValues (0, 0.1, 0.2, ... , 1)
         for r in recallValues:
             # Obtain all recall values higher or equal than r
-            argGreaterRecalls = np.argwhere(mrec[:-1] >= r)
+            argGreaterRecalls = np.argwhere(mrec[:] >= r)
             pmax = 0
             # If there are recalls above r
             if argGreaterRecalls.size != 0: