使用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