神经网络的ROC曲线

时间:2020-06-19 03:23:17

标签: python roc cnn

我正在尝试绘制roc曲线,以解决使用CNN模型训练的二进制图像分类问题。我使用了以下python代码,但导致错误。命令predict_proba(x_test)没有显示属性错误。您能建议我正确的代码顺序来解决此问题吗?谢谢

    predictions = cnn_model.predict_proba(X_test)
    fpr1, tpr1, _ = roc_curve(y_test, predictions[:,1])
    plt.plot(fpr1, tpr1)
    plt.xlabel('FPR')
    plt.ylabel('TPR')
    plt.title('ROC curve')
    plt.show()

0 个答案:

没有答案
相关问题