我使用 Keras 2.2.2,tensorflow 1.9.0和python3.5 构建了一个二进制图像分类器。
我得到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
,它都应该显示出两类正确的概率。像这样
为什么我得到单个概率值?