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)