ROC 曲线的尺寸不正确

Question

我是机器学习的新手，负责创建ROC曲线。我得到错误ValueError:Found输入变量的样本数不一致：[200400]。这是我的ROC曲线代码的图片。

import sklearn.metrics as metrics
from sklearn.metrics import roc_curve, auc
y_pred = basic_model.predict(X_test).ravel()
nn_fpr_keras, nn_tpr_keras, nn_thresholds_keras = roc_curve(y_test, y_pred)
auc_keras = auc(nn_fpr_keras, nn_tpr_keras)
plt.plot(nn_fpr_keras, nn_tpr_keras, marker='.', label='Neural Network (auc = %0.3f)' % auc_keras)

事实上，我知道问题出在哪里——我用"稀疏分类交叉熵"来表示损失。(我的顾问说我必须用它(。基本上，他们还告诉我，现在我所要做的就是将X_test从2D数组更改为1D数组。我试过用.flatin((，但没用？关于我应该做什么，或者我可以研究的资源，有什么想法吗？

Answer 1

我只需要设置X_train=X_train[；，0]Ravel((似乎不起作用。