检查目标时出错:预期density_1具有3维,但数组的形状为(118,1)

检查目标时出错:预期density_1具有3维,但数组的形状为(118,1),第1张

检查目标时出错:预期density_1具有3维,但数组的形状为(118,1)

您的第二个LSTM层还返回序列,默认情况下,Dense层将内核应用于每个时间步并生成一个序列:

# (bs, 45, 2)model.add( LSTM( 512, input_shape=(45, 2), return_sequences=True))# (bs, 45, 512)model.add( LSTM( 512, return_sequences=True))# (bs, 45, 512)model.add( (Dense(1)))# (bs, 45, 1)

因此,您的输出就是shape

(bs, 45,1)
。要解决该问题,您需要
return_sequences=False
在第二个LSTM层中进行设置,以压缩序列:

# (bs, 45, 2)model.add( LSTM( 512, input_shape=(45, 2), return_sequences=True))# (bs, 45, 512)model.add( LSTM( 512, return_sequences=False)) # SET HERE# (bs, 512)model.add( (Dense(1)))# (bs, 1)

然后,您将获得所需的输出。注意

bs
是批次大小。



欢迎分享,转载请注明来源:内存溢出

原文地址:https://www.54852.com/zaji/5639975.html

(0)
打赏 微信扫一扫微信扫一扫 支付宝扫一扫支付宝扫一扫
上一篇 2022-12-16
下一篇2022-12-16

发表评论

登录后才能评论

评论列表(0条)

    保存