1. 程式人生 > >scikit-learn常用介面

scikit-learn常用介面

1)roc_auc_score 獲取AUC score

用法:

from sklearn import metrics auc = metrics.roc_auc_score(y_true, y_score) 常見問題:

Data is not binary and pos_label is not specified

原因:y_true必須是0 1 array,如果不是0,1可以有兩種方式解決。

方式一:將y_true轉為0,1 array

方式二:y_true可以不是0,1 但是需要指定pos_label

fpr, tpr, thresholds = metrics.roc_curve(y_true, y_score, pos_label='T')

auc = metrics.auc(fpr, tpr)