【发布时间】:2017-01-03 09:56:15
【问题描述】:
我正在尝试从 Keras 上的 SimpleRNN 上的文本进行训练。
在 Keras 中,我为 SimpleRNN 指定了一个非常简单的参数,如下所示:
model = Sequential()
model.add(SimpleRNN(output_dim=1, input_shape=(1,1,1))
我明白 input_shape 应该是 (nb_samples, timesteps, input_dim),和我的 train_x.shape 一样
所以我很惊讶我收到了以下错误。
Traceback (most recent call last):
File "C:/Users/xxx/xxxx/xxx/xxx.py", line 262, in <module>
model.add(SimpleRNN(output_dim=vocab_size, input_shape=train_x.shape))
File "C:\Anaconda3\envs\py34\lib\site-packages\keras\models.py", line 275, in add
layer.create_input_layer(batch_input_shape, input_dtype)
File "C:\Anaconda3\envs\py34\lib\site-packages\keras\engine\topology.py", line 367, in create_input_layer
self(x)
File "C:\Anaconda3\envs\py34\lib\site-packages\keras\engine\topology.py", line 467, in __call__
self.assert_input_compatibility(x)
File "C:\Anaconda3\envs\py34\lib\site-packages\keras\engine\topology.py", line 408, in assert_input_compatibility
str(K.ndim(x)))
Exception: Input 0 is incompatible with layer simplernn_1: expected ndim=3, found ndim=4
不知道为什么 keras 在只指定 3 时“发现 ndim=4”!
为了清楚起见,我的
train_x.shape = (73, 84, 400)
和
vocab_size=400
。只要输入 3d 及以上的 input_shape,我意识到会导致错误。
任何帮助将不胜感激!!! :))
【问题讨论】:
标签: python neural-network theano keras recurrent-neural-network