Keras重量文件加载例外:将2层加载到具有0层的模型中

时间:2018-10-16 23:25:28

标签: python machine-learning neural-network keras dropout

当我在输入层添加辍学时发生异常。

在其他线程中也提到了该异常,它与另一个问题有关,最常见的建议解决方案是降级Keras版本。有此异常的解决方法吗?

def baseline_model() :
    model = Sequential()
    model.add(Dropout(0.35)) #THIS LINE CAUSES THE EXCEPTION
    model.add(Dense(200, input_dim=1200, kernel_initializer='normal', activation='relu'))
    model.add(Dropout(0.8))
    rms = RMSprop(lr = 0.00050)
    model.add(Dense(1, kernel_initializer='normal', activation='sigmoid'))
    model.compile(loss='binary_crossentropy', optimizer=rms, metrics=['accuracy'])
    return model

模型在加载重量文件时引发以下异常:

ValueError: You are trying to load a weight file containing 2 layers into a model with 0 layers.

1 个答案:

答案 0 :(得分:0)

问题是您尚未为第一层分配输入形状:

model.add(Dropout(0.35, input_shape=(1200,)))

然后删除第二层的input_dim参数,因为它是多余的。