Average Precision

ap.py

def ap_k(res: list[bool], K: int):
    out = np.zeros(K, dtype=np.float32)
    rel = np.zeros(K, dtype=np.float32)
    
    for k in range(K):
        rel[k] = res[k] # is current result relevant?
        out[k] = sum(rel) / (k + 1) # how many relevant results until now?

    if rel.sum() == 0:
        return 0
      
    return (out * rel).sum() / rel.sum()