5

我对卷积层和 lstm 层之间的连接有疑问。数据的形状为 (75,5),其中每个时间步有 75 个时间步 x 5 个数据点。我想做的是对(75x5)进行卷积,获取新的卷积(75x5)数据并将该数据输入 lstm 层。但是,它不起作用,因为卷积层的输出形状具有我不需要的过滤器数量。因此卷积层输出的形状为 (1,75,5),lstm 层所需的输入为 (75,5)。我如何只取第一个过滤器。

model = Sequential() 
model.add(Convolution2D(1, 5,5,border_mode='same',input_shape=(1,75, 5)))
model.add(Activation('relu'))
model.add(LSTM(75, return_sequences=False, input_shape=(75, 5)))
model.add(Dropout(0.5))
model.add(Dense(1))
model.compile(loss='mse', optimizer='rmsprop')

这是出现的错误:

File "/usr/local/lib/python3.4/dist-packages/keras/layers/recurrent.py", line 378, in __init__
super(LSTM, self).__init__(**kwargs)
File "/usr/local/lib/python3.4/dist-packages/keras/layers/recurrent.py", line 97, in __init__
super(Recurrent, self).__init__(**kwargs)
File "/usr/local/lib/python3.4/dist-packages/keras/layers/core.py", line 43, in __init__
self.set_input_shape((None,) + tuple(kwargs['input_shape']))
File "/usr/local/lib/python3.4/dist-packages/keras/layers/core.py", line 138, in set_input_shape
', was provided with input shape ' + str(input_shape))
Exception: Invalid input shape - Layer expects input ndim=3, was provided with input shape (None, 1, 75, 5)
4

1 回答 1

5

您可以在两者之间添加 Reshape() 层以使尺寸兼容。

http://keras.io/layers/core/#reshape

keras.layers.core.Reshape(dims)

将输出重塑为特定形状。

输入形状

任意的,尽管输入形状中的所有维度都必须是固定的。将此层用作模型中的第一层时,请使用关键字参数 input_shape(整数元组,不包括样本轴)。

输出形状

(batch_size,) + dims

论据

dims: 目标形状。整数元组,不包括样本维度(批量大小)。

于 2016-02-07T15:14:12.753 回答