python计算AUC

关注B站可以观看更多实战教学视频：hallo128的个人空间

python计算AUC

使用的是sklearn中的roc_auc_score () 方法

核心代码

from sklearn.metrics import roc_auc_score
train_label_all = p2['target']   # 真实的标签
train_prob1_all = p_test['y_pred']  # 预测标签
print("AUC:{:.4f}".format(roc_auc_score(train_label_all, train_prob1_all)))

运行效果演示

要保证真实标签和预测标签是一一对应的