peanutwang
peanutwang
采纳率0%
2019-06-12 11:22 阅读 295

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条回答 默认 最新

  • peanutwang peanutwang 2019-06-12 14:44

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

    点赞 评论 复制链接分享

相关推荐