Как передать вход в правильную форму слою вложения Keras? - PullRequest
Новая
       110

Как передать вход в правильную форму слою вложения Keras?

0 голосов
/ 28 апреля 2020

Я занимаюсь разработкой программы LSTM для проблемы НЛП. Форма моих данных и метки: = (10,20,1)

Код моей модели выглядит следующим образом: 

model.add(Embedding(18,17,input_length=20,weights=[embedding_weights])) ( Shape of Embedding (18,17))
# encoder layer
model.add(LSTM(100, activation='relu', input_shape=(20, 1)))

# repeat vector
model.add(RepeatVector(20))

# decoder layer
model.add(LSTM(100, activation='relu', return_sequences=True))
model.add(TimeDistributed(Dense(1)))
model.compile(optimizer='adam', loss='mse')

Я получаю следующую ошибку

"input_length "20, но полученный вход имеет форму (Нет, 20, 1)

...