错误的模型预测

Question

我有一个二元分类问题。我想检测图像上的雨滴。我训练了一个简单的模型，但我的预测并不好。我想在 0 到 1 之间进行预测。

对于我的第一次尝试，我使用 relu 表示所有层都接受最终（我使用了 softmax）。作为优化器，我使用了binary_crossentropy并将其更改为categorical_crossentropy。他们两个都没有用。

opt = Adam(lr=LEARNING_RATE, decay=LEARNING_RATE / EPOCHS)
cnNetwork.compile(loss='categorical_crossentropy',
              optimizer=optimizers.RMSprop(lr=lr),
              metrics=['accuracy']) 

inputShape = (height, width, depth)
    # if we are using "channels first", update the input shape
    if K.image_data_format() == "channels_first":
        inputShape = (depth, height, width)
    # First layer is a convolution with 20 functions and a kernel size of 5x5 (2 neighbor pixels on each side)
    model.add(Conv2D(20, (5, 5), padding="same",
        input_shape=inputShape))
    # our activation function is ReLU (Rectifier Linear Units)
    model.add(Activation("relu"))
    # second layer is maxpooling 2x2 that reduces our image resolution by half 
    model.add(MaxPooling2D(pool_size=(2, 2), strides=(2, 2)))
    # Third Layer - Convolution, twice the size of the first convoltion
    model.add(Conv2D(40, (5, 5), padding="same"))
    model.add(Activation("relu"))
    model.add(MaxPooling2D(pool_size=(2, 2), strides=(2, 2)))
    # Fifth Layer is Full connected flattened layer that makes our 3D images into 1D arrays
    model.add(Flatten())
    model.add(Dense(500))
    model.add(Activation("relu"))
    # softmax classifier
    model.add(Dense(classes))
    model.add(Activation("softmax"))

我希望第一类得到

ex.1，第二类得到 .9。结果，我得到 1 ， 1.3987518e-35。主要问题是我总是得到 1 作为预测。

Answer 1

你应该使用binary_crossentropy，你得到的输出没有任何问题。输出 1 ， 1.3987518e-35 表示一等概率几乎为 1，二等概率非常接近 0 （1e-35）。