如何在Keras中标记CNN并使用MSE指标

Question

我必须使用损失函数mse训练cnn。 我将标签标准化为0-1，但训练时我得到了

  

期望density_2的形状为（1，），但数组的形状为（7，）

最后一层是 x = Dense(1, activation='linear')(x)

当我编译和训练时：

model.compile(loss=keras.losses.mean_squared_error, optimizer=Adam(lr=0.001),metrics=['mse'])


model_details=model.fit(X_train, Y_train)

Answer 1

我将Y_train矩阵的大小调整为一维向量：向量的每个分量都是相应图像的标签。