使用sklearn严格计算AUROC和AUPRC

使用sklearn中的方法严格计算AUROC和AUPRC的值

1. AUROC的计算方法:

from sklearn.metrics import roc_curve, auc
fpr, tpr, _ = roc_curve(y_true, y_score)
auroc = auc(fpr, tpr)

注意:其中y_score指的是模型输出的预测概率值,不是预测标签!

2. AUPRC的计算方法:

from sklearn.metrics import precision_recall_curve, auc
precision, recall, _ = precision_recall_curve(y_true, y_score, pos_label=1)
auprc = auc(recall, precision)

注意:和上面一样,y_score指的也是模型输出的概率值;pos_label指的是正样本标签,默认为1

你可能感兴趣的:(机器学习,sklearn,python)