keras - 如何在 keras 中标记 cnn 并使用 mse 指标

Question

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

预期dense_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)

标签： kerasconv-neural-networkmse

Answer 1

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