Precision-Recall Curve feature update #1206

This commit is contained in:
Glenn Jocher authored and GitHub committed 2020-10-25 12:59:16 +01:00
1 parent ed85038c38
commit 8e9b3d0205
1 file changed
+1 -1
+1 -1
View File
@@ -337,7 +337,7 @@ def compute_ap(recall, precision):
# Integrate area under curve
method = 'interp' # methods: 'continuous', 'interp'
if method == 'interp':
x = np.linspace(0, 1, 1001) # 101-point interp (COCO)
x = np.linspace(0, 1, 101) # 101-point interp (COCO)
ap = np.trapz(np.interp(x, mrec, mpre), x) # integrate
else: # 'continuous'
i = np.where(mrec[1:] != mrec[:-1])[0] # points where x axis (recall) changes