【发布时间】:2019-10-02 17:05:13
【问题描述】:
训练输入形状:(13974, 100, 6, 5)
训练输出形状:(13974, 1,1)
测试输入形状:(3494, 100, 6, 5)
测试输出形状:(3494, 1, 1)
我正在开发以下模型。 2D CNN LSTM。
model = Sequential()
model.add(TimeDistributed(Conv2D(1, (1,1), activation='relu',
input_shape=(6,5,1))))
model.add(TimeDistributed(MaxPooling2D(pool_size=(6, 5))))
model.add(TimeDistributed(Flatten()))
model.add(LSTM(units=300, return_sequences= False, input_shape=(100,1)))
model.add(Dense(1))
当我尝试按照以下方式进行调整时
model.fit(train_input,train_output,epochs=50,batch_size=60)
它给了我一个错误。
ValueError: strides 的长度应为 1、1 或 3,但为 2
请更正我的模型。我正在将 6,5 图像转换为单个单元,并从 100 个时间戳中预测第 101 个时间戳。
【问题讨论】:
标签: image-processing keras conv-neural-network lstm keras-layer