【发布时间】: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