私信  •  关注

Xin Niu

Xin Niu 最近回复了
5 年前
回复了 Xin Niu 创建的主题 » Python逻辑回归中scikit学习中的巨大而奇怪的错误?

我认为错误发生是因为预测只有一行, 有两个列名:

prediction=pd.DataFrame(data=predictions, 
                         columns=['Prob of Bad credit (0)','Prob of Good credit (1)'])

y_pred_prob=logreg.predict_proba(x_test)[:,:]
y_pred_prob_df=pd.DataFrame(data=y_pred_prob, columns=['Prob of no heart disease (0)','Prob of Heart Disease (1)'])
y_pred_prob_df.head()

我想你应该把代码改成:

prediction_df = pd.DataFrame(data=prediction,  
                         columns=['Prob of Bad credit (0)','Prob of Good credit (1)'])

小心,应该是预测,而不是预测。