График RO C кривая модели ANN - PullRequest
Новая
       63

График RO C кривая модели ANN

1 голос
/ 19 июня 2020

Я работаю над своей моделью ANN и пытаюсь построить график результатов RO C.

Мой ввод для кода RO C - это y_test и прогнозы.

Как выглядит Y_test: 

[0.5875     0.48229167 0.58125    ... 0.54375    0.60729167 0.628125  ]

Как выглядят прогнозы: 

[[0.63529354]
 [0.5447516 ]
 [0.5529088 ]
 ...
 [0.5529088 ]
 [0.5894695 ]
 [0.5880237 ]]

Часть кода после оценки модели: 

predictions = model.predict(X_test)

# Compute ROC curve and ROC area for each class
fpr = dict()
tpr = dict()
roc_auc = dict()

for i in range (len(Y_test)):
    fpr[i], tpr[i], _ = roc_curve(Y_test[i], predictions[:,i])
    roc_auc[i] = auc(fpr[i], tpr[i])

# Compute micro-average ROC curve and ROC area
fpr["micro"], tpr["micro"], _ = roc_curve(Y_test.ravel(), hist.ravel())
roc_auc["micro"] = auc(fpr["micro"], tpr["micro"])

Когда я попробуйте запустить код, я получаю сообщение об ошибке: 

    raise ValueError("{0} format is not supported".format(y_type))
ValueError: continuous format is not supported

Ошибка касается строки и элемента Y_test [i]: 

`    fpr[i], tpr[i], _ = roc_curve(Y_test[i], predictions[:,i])`

В чем моя проблема?

...