Всегда получая AUC = 0,5 - PullRequest
Новая
       8

Всегда получая AUC = 0,5

0 голосов
/ 06 января 2019

Я пытаюсь использовать сверточную нейронную сеть (CNN) для прогнозирования классов тестовых изображений следующим образом: 

for root, dirs, files in os.walk(test_directory):
    for file in files:
        img = cv2.imread(root + '/' + file)
        img = cv2.resize(img,(512,512),interpolation=cv2.INTER_AREA)
        img = np.expand_dims(img, axis=0)
        img = img/255.0
        if os.path.basename(root) == 'nevus':
            label = 1
        elif os.path.basename(root) == 'melanoma':
            label = 0
        labels.append(label)
        img_class = model.predict_classes(img)
        img_class_probability = model.predict(img)
        prediction_probability = img_class_probability[0]
        prediction_probabilities.append(prediction_probability)
        prediction = img_class[0]
        if prediction == label:
            correct_classification = correct_classification + 1
        number_of_test_images = number_of_test_images + 1

fpr, tpr, thresholds = roc_curve(labels, prediction_probabilities)
auc_value = auc(fpr, tpr)

Я хотел бы спросить, почему у меня всегда AUC = 0,5? Может я что-то не так делаю?

Спасибо.

...