Keras的网格数据时间序列预测

Question

我正在尝试预测5天的降水量，但是该模型似乎并没有在某种程度上起作用。我正在使用convLSTM。我的输入是5维的（批次，时间步，纬度，经度，通道）。

seq = Sequential()
seq.add(ConvLSTM2D(filters=40,kernel_size=(3,3),input_shape=(None,50, 40, 1), padding='same', return_sequences=True))

seq.add(BatchNormalization())

seq.add(ConvLSTM2D(filters=40, kernel_size=(3, 3),
               padding='same', return_sequences=True))
seq.add(BatchNormalization())

seq.add(Lambda(lambda x: x[: ,-5: ,: ,: ,:]))

seq.add(Conv3D(filters=1, kernel_size=(3, 3, 3),
           activation='sigmoid',
           padding='same', data_format='channels_last'))

我有点困惑。我认为架构可以做到这一点，

                     0 0 0 0 0
           | | | | | | | | | |
           0 0 0 0 0 0
           | | | | | |
       ....0 0 0 0 0 0

对吗？ 如果没有，建议更改。