Кривая RO C для нейронных сетей - PullRequest
Новая
       71

Кривая RO C для нейронных сетей

0 голосов
/ 19 июня 2020

Я пытаюсь построить кривую ro c для задачи классификации двоичных изображений, обученной с использованием модели CNN. Я использовал следующие коды python, но это привело к ошибке. Команда expect_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()
...