这个错误提示意思是输入的变量样本数量不一致,其中y_train有22397个样本,而pred_test有2001个样本。这是因为在计算AUC值时,应该使用训练集的预测结果pred_train而不是测试集的预测结果pred_test。

将auc_tra = roc_auc_score(y_train, pred_test)修改为auc_tra = roc_auc_score(y_train, clf.predict_proba(X_train)[:, 1])即可。


原文地址: https://www.cveoy.top/t/topic/erEx 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录