gpt4 book ai didi

neural-network - 在 keras 或 Tensorflow 中的 LSTM 层之前添加密集层?

转载 作者:行者123 更新时间:2023-12-04 06:43:33 29 4
gpt4 key购买 nike

我正在尝试实现一个去噪自动编码器,中间有一个 LSTM 层。架构如下。

FC layer -> FC layer -> LSTM cell -> FC layer -> FC layer.

我无法理解我的输入维度应该如何实现这个架构?

我尝试了以下代码
batch_size = 1
model = Sequential()
model.add(Dense(5, input_shape=(1,)))
model.add(Dense(10))
model.add(LSTM(32))
model.add(Dropout(0.3))
model.add(Dense(5))
model.add(Dense(1))
model.compile(loss='mean_squared_error', optimizer='adam')
model.fit(trainX, trainY, nb_epoch=100, batch_size=batch_size, verbose=2)

我的 trainX 是 [650,20,1] 向量。它是一个只有一个特征的时间序列数据。

我收到以下错误
ValueError                                Traceback (most recent call last)
<ipython-input-20-1248a33f6518> in <module>()
3 model.add(Dense(5, input_shape=(1,)))
4 model.add(Dense(10))
----> 5 model.add(LSTM(32))
6 model.add(Dropout(0.3))
7 model.add(Dense(5))

/usr/local/lib/python2.7/dist-packages/keras/models.pyc in add(self, layer)
330 output_shapes=[self.outputs[0]._keras_shape])
331 else:
--> 332 output_tensor = layer(self.outputs[0])
333 if isinstance(output_tensor, list):
334 raise TypeError('All layers in a Sequential model '

/usr/local/lib/python2.7/dist-packages/keras/engine/topology.pyc in __call__(self, x, mask)
527 # Raise exceptions in case the input is not compatible
528 # with the input_spec specified in the layer constructor.
--> 529 self.assert_input_compatibility(x)
530
531 # Collect input shapes to build layer.

/usr/local/lib/python2.7/dist-packages/keras/engine/topology.pyc in assert_input_compatibility(self, input)
467 self.name + ': expected ndim=' +
468 str(spec.ndim) + ', found ndim=' +
--> 469 str(K.ndim(x)))
470 if spec.dtype is not None:
471 if K.dtype(x) != spec.dtype:

ValueError: Input 0 is incompatible with layer lstm_10: expected ndim=3, found ndim=2

最佳答案

密集层可以将序列作为输入,它将在每个向量(最后一个维度)上应用相同的密集层。例子 :

您有一个表示序列 (timesteps, dim_features) 的二维张量输入,如果您使用 new_dim 输出对其应用密集层,则在该层之后您将拥有的张量将是一个新序列 (timesteps, new_dim)
如果你有一个 3D 张量 (n_lines, n_words, embedding_dim)可以是文档,用 n_lines行,n_words每行字数和 embedding_dim每个单词的维度,使用 new_dim 输出对其应用密集层将为您提供形状为 (n_lines, n_words, new_dim) 的新文档张量(3D)

您可以 see here您可以使用 Dense() 层提供和获取的维度输入和输出。

关于neural-network - 在 keras 或 Tensorflow 中的 LSTM 层之前添加密集层?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/42714688/

29 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com