2019年10月28日
sklearn里计算roc_auc_score，报错ValueError: bad input shape
摘要： 用sklearn的DecisionTreeClassifer训练模型，然后用roc_auc_score计算模型的auc。代码如下 报错信息如下 目测是你的y_pred出了问题，你的y_pred是(900, 2)的array，也就是有两列。 因为predict_proba返回的是两列。predict_ 阅读全文
posted @ 2019-10-28 08:49 anovana 阅读(2757) 评论(0) 推荐(1) 编辑