AUC，准确率（ACC），敏感性（sensitivity），特异性（specificity）计算 Python

Posted 2022-01-15 K同学啊

import numpy as np
from sklearn.metrics import confusion_matrix,accuracy_score,f1_score,roc_auc_score,recall_score,precision_score
 
def calculate_metric(gt, pred): 
    pred[pred>0.5]=1
    pred[pred<1]=0
    confusion = confusion_matrix(gt,pred)
    TP = confusion[1, 1]
    TN = confusion[0, 0]
    FP = confusion[0, 1]
    FN = confusion[1, 0]
    print('Accuracy:',(TP+TN)/float(TP+TN+FP+FN))
    print('Sensitivity:',TP / float(TP+FN))
    print('Specificity:',TN / float(TN+FP))  
               
y = []
pred = []

auc = roc_auc_score(y,pred)
print('AUC:',auc)
calculate_metric(y,pred)