小编Jan*_*rak的帖子

为什么我的模型始终获得 0.5 AUC?

我目前正在做一个项目,需要在一组图像中预测眼部疾病。我正在使用 Keras 内置应用程序。我在 VGG16 和 VGG19 上得到了很好的结果,但在 Xception 架构上,我每个时期的 AUC 都恰好为 0.5。

我尝试了不同的优化器和学习率,但没有任何效果。我通过从 RMSProp 优化器切换到 Adam 优化器解决了 VGG19 的相同问题,但我无法让它适用于 Xception。

   def buildModel():
    from keras.models import Model
    from keras.layers import Dense, Flatten
    from keras.optimizers import adam

    input_model = applications.xception.Xception(
        include_top=False,
        weights='imagenet',
        input_tensor=None,
        input_shape=input_sizes["xception"],
        pooling=None,
        classes=2)

    base_model = input_model

    x = base_model.output
    x = Flatten()(x)
    predictions = Dense(2, activation='softmax')(x)

    model = Model(inputs=base_model.input, outputs=predictions)
    for layer in base_model.layers:
        layer.trainable = False

    model.compile(optimizer=adam(lr=0.01), loss='binary_crossentropy', metrics=['accuracy'])

    return model


class Histories(keras.callbacks.Callback):

    def __init__(self, val_data):
        super(Histories, self).__init__() …
Run Code Online (Sandbox Code Playgroud)

python scikit-learn auc deep-learning keras

3
推荐指数
1
解决办法
3889
查看次数

标签 统计

auc ×1

deep-learning ×1

keras ×1

python ×1

scikit-learn ×1