关于sklearn的roc_auc_score

占坑记录:

跑模型出分数的时候被这个方法差点坑了,怪自己理解不够到位,占坑记录一下。

最早发现这个问题是发觉输入roc_auc_score(xtest,prediction)输出的auc与plot_auc的值相差甚远,查资料之后发现关键在于第二个参数应该输入模型的输出概率值而非预测值,这里是个小坑 ,因为其他如acc pre f1score都是输入真实与预测值,当模型输出概率时,应该输入Prob。

你可能感兴趣的:(个人常用,碎碎念,深度学习,sklearn,python,机器学习)