我认为错误发生是因为预测只有一行,
有两个列名:
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)'])
小心,应该是预测,而不是预测。