如何解决AxisError:轴 1 超出 ROC AUC 上第 1 维数组的范围
我正在尝试使用分层 k 折和多项逻辑回归来拆分数据集,但是在使用此代码时出现了上述错误:
i=1
for train_set,test_set in skf.split(x,y):
print('{} of SKFold {}'.format(i,skf.n_splits))
xtr,xvl = x.iloc[train_set],x.iloc[test_set]
ytr,yvl = y.iloc[train_set],y.iloc[test_set]
#model
model.fit(xtr,ytr)
score = roc_auc_score(yvl,model.predict(xvl),multi_class='ovr')
print('ROC AUC score:',score)
cv_score.append(score)
pred_test = model.predict_proba(x_test)[:,1]
pred_test_full +=pred_test
i+=1
'model' 的值如下:
model = LogisticRegression(random_state=0,multi_class='multinomial',penalty='none',solver='newton-cg').fit(x_train,y_train)
我该怎么做才能解决这个问题?
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。