Precision-Recall Curve feature update #1206
This commit is contained in:
1 parent
ed85038c38
commit
8e9b3d0205
1 file changed
+1
-1
+1
-1
@@ -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
|
||||
|
||||
Reference in new issue
Block a user