CNN + LSTM模型

时间:2019-10-01 10:28:38

标签: python tensorflow conv-neural-network lstm faster-rcnn

\@
  

13974-序列100-时间戳6-位置5-变量

我需要我的模型预测6个位置的下一个101时间戳,每个位置5个变量。

train input shape :  (13974, 100, 6, 5) 
train output shape :   (13974, 1, 6, 5) 
test input shape :   (3494, 100, 6, 5) 
test output shape :   (3494, 1, 6, 5)

这是我的模型,出现错误ValueError:步幅应为1、1或3,但长度为2

我需要喂火车model = Sequential() model.add(TimeDistributed(Conv2D(32, 5, 5,input_shape=(100,6,5,1)))) model.add(TimeDistributed(MaxPooling2D(pool_size=(4, 4)))) model.add(Activation('relu')) model.add(TimeDistributed(Flatten())) model.add(LSTM(100, return_sequences=True)) model.add(TimeDistributed(Dense(5))) model.compile(loss='mae', optimizer='adam', metrics=['accuracy']) ,并预测火车o input : (13974, 100, 6, 5)。请更正我的模型和必要的更正吗?

0 个答案:

没有答案