填最后一个坑,如何寻找误判的病例。
之前我们在介绍AUC的时候,提到了两个函数:predict和predict_proba,复习一下:
auc_test = roc_auc_score(y_test, y_testprba)
roc_auc_score的参数呢,包括两个:y_test是实际值,y_testprba是预测的概率(注意,是概率,而不是分类,要和y_pred做区别),来看看代码:
y_pred = classifier.predict(X_test)
y_testprba = classifier.predict_proba(X_test)[:,1]