Я построил двоичный классификатор изображений с 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
, это должно показывать вероятность для двух классов правильно.Вот так
Почему я получаю единичные значения вероятности?