【问题标题】:How label a cnn in keras and use mse metrics如何在 keras 中标记 cnn 并使用 mse 指标
【发布时间】:2019-05-18 09:32:39
【问题描述】:

我必须使用损失函数 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)

【问题讨论】:

  • 你能包含你的代码吗?或者至少是您构建模型的部分?
  • 什么是Y_train.shape

标签: keras conv-neural-network mse


【解决方案1】:

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

【讨论】:

    猜你喜欢
    • 2020-10-03
    • 2018-02-07
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2021-01-24
    相关资源
    最近更新 更多