import pandas as pd from sklearn import metrics from sklearn.metrics import auc # import argparse def get_roc_curve(y,pred,pos_label): """ 计算二分类问题的roc和auc """ fpr, tpr, thresholds = metrics.roc_curve(y, pred, pos_label) AUC = metrics.auc(fpr, tpr) print(AUC)