Pārlūkot izejas kodu

Precision-Recall Curve feature update #1206

5.0
Glenn Jocher GitHub pirms 3 gadiem
vecāks
revīzija
8e9b3d0205
Šim parakstam datu bāzē netika atrasta zināma atslēga GPG atslēgas ID: 4AEE18F83AFDEB23
1 mainītis faili ar 1 papildinājumiem un 1 dzēšanām
  1. +1
    -1
      utils/general.py

+ 1
- 1
utils/general.py Parādīt failu

@@ -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

Notiek ielāde…
Atcelt
Saglabāt