Почему существует только один столбец, содержащий значения вероятности при прогнозировании на изображениях (двоичная классификация изображений)? - PullRequest
0 голосов
/ 01 декабря 2018
  • Я построил двоичный классификатор изображений с Keras 2.2.2, тензор потока 1.9.0 и python3.5 .

  • IПолучаю training accuracy:88% и validation accuracy:83%. Также, AUC:.803.

  • Обученные изображения на 2000 изображений для обоих классов.

Моя структура каталогов тестирования:

  • тест:
    • меланома
    • невус

вот код тестирования.

#loading the test data

valid_path = '/media/centura/UUI/danish/skincancer/testing1/test/'

test_gen = datagen.flow_from_directory(valid_path,
                                        target_size=(IMAGE_SIZE,IMAGE_SIZE),
                                        batch_size=1,
                                        class_mode='binary',
                                        shuffle=False)

y_pred_keras = model.predict_generator(test_gen,steps=70, verbose=1)

печатьy_pred_keras

[[.4235]
[.4213]
[.4367]
[.418]]

Как правило, когда берется y_pred, это должно показывать вероятность для двух классов правильно.Вот так

table1

Почему я получаю единичные значения вероятности?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...