时间步长问题。tensorflow训练lstm时序模型,输出层实际输出维度和期待维度不一致

时间:2024-04-24 19:04:27

设置输出维度为1.
Dense(1)
但结果跑出来的输出维度每次都是三维的。
模型设置:

模型构建

模型输出

输入x维度(2250,48,2)
输入y 维度(2250,) 和 (2250,1)
但模型预测出的结果维度都是(2250,48,1)

我就很纳闷= = !
后来对比了以前跑过的文件。发现LSTM少定义一个参数。
在这里插入图片描述

input_shape填两个参数值,第一个值代表指定的时间步长。此处为48.
第二个值是特征数目。
更改后训练成功。模型预测值维度为(2250,1)
在这里插入图片描述
问题解决~