0

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

1 回答 1

0

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

于 2019-05-18T15:57:54.917 回答