从接触机器学习到现在,在做二分类项目时,一直都用的
y_pred = model.predict(X_test)
auc = roc_auc_score(y_true, y_pred)
到今天才发现原来用错了,用y_pred = model.predict_proba(X_test)[:, 1]
比model.predict(X_test)
的auc值要高很多
可以参考sk-learn
的roc_auc_score
函数文档
07-01
3万+