Keras做序列到序列任务,出现这样的低级错误该怎么解决?

Keras 处理一个序列到序列问题,输入序列如下:
图片说明

输出序列如下:

图片说明

它们的shape是(55,60)
即共55句,每句60个word

模型如下:
model = Sequential()
model.add(Embedding(MAX_NB_WORDS, EMBEDDING_DIM, mask_zero=True,input_shape=(MAX_SEQUENCE_LENGTH,))) # Random embedding
model.add(Bidirectional(LSTM(BiRNN_UNITS // 2, return_sequences=True)))
model.add(Dense(2,activation='softmax'))
model.compile(optimizer='rmsprop', loss='categorical_crossentropy', metrics=['acc'])

模型summary如下:
图片说明

出现这样的错误,请问该怎么解决?
Error when checking target: expected dense_1 to have 3 dimensions, but got array with shape (45, 60)

1个回答

问题已经得到解决。原因是Dense做激活层的时候,需要把维度变换一下。

Csdn user default icon
上传中...
上传图片
插入图片
抄袭、复制答案,以达到刷声望分或其他目的的行为,在CSDN问答是严格禁止的,一经发现立刻封号。是时候展现真正的技术了!
立即提问