auc_pr_roc.py 文件源码

def auc_pr(real_csv, result_csv):
    '''??real.csv?result.csv??????PR???AUC?'''
    label, prob = load_label_prob(real_csv, result_csv)
    precision, recall, _thresholds = metrics.precision_recall_curve(label, prob)
    area = metrics.auc(recall, precision)
    #print(area)
    return area