keras模型加载错误

Question

我使用

成功保存了模型后训练

model.save("my_model.h5")

现在，当我尝试使用load_model加载模型时，它会出现以下错误

model = load_model("my_model.h5") 
ValueError: You are trying to load a weight file containing 8 layers into a model with 0 layers. 

我的代码：

model = create_model(layer_sizes1, layer_sizes2, input_shape1, input_shape2,learning_rate, reg_par, outdim_size, use_all_singular_values)
model.summary()
model = train_model(model, data1, data2, epoch_num, batch_size)
model.save("my_model.h5")

这会保存模型，但是当我尝试load_model时会出现上述错误。

模型是具有三个隐藏层的神经网络。您可以在此处找到模型定义：https://github.com/adakum/DeepCCA

Answer 1

要解决此问题，您需要在第一层中直接添加input_shape=(X.shape[1],)。例如，您的X的形状是10行100列。您需要将input_shape=(100,)添加到第一层