为什么对图像进行预测(二进制图像分类)时只有一列包含概率值?

时间:2018-12-01 11:45:38

标签: python-3.x tensorflow conv-neural-network prediction

  • 我使用 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,它都应该显示出两类正确的概率。像这样

table1

为什么我得到单个概率值?

0 个答案:

没有答案