diff --git a/lib/Evaluator.py b/lib/Evaluator.py index b44d30fe..ae9874cc 100644 --- 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: