def median_cvtest(cv_results): auprs = [(i, cv.aupr(c.ppis, c.ntest_pos)) for i,c in enumerate(cv_results)] auprs.sort(key=lambda x: x[1]) median_index = auprs[int(len(auprs)/2)][0] return cv_results[median_index]
def aupr(gold, ppis): """ Area under precision-recall. """ tested,ntest_pos = tested_ppis(gold, ppis) return cv.aupr(tested, ntest_pos)