【发布时间】:2019-11-20 06:28:07
【问题描述】:
X_train 数据维度为 87482, 3。
但我在使用以下代码运行时出错。错误是:
ValueError: 检查输入时出错:预期 lstm_6_input 的形状为 (87482, 1) 但得到的数组的形状为 (87482, 3)
我的代码是:
model = Sequential()
#model.add(Embedding(top_words, embedding_vecor_length, input_length=max_review_length))
model.add(LSTM(units=3, input_shape=(X_train_rnn.shape[1],1),return_sequences=True))
model.add(LSTM(3, return_sequences=True)) # returns a sequence of vectors of dimension 32
model.add(LSTM(3))
model.compile(loss='mean_squared_error', optimizer='adam', metrics=['accuracy'])
print(model.summary())
model.fit(X_train_rnn, y_train, epochs=2, batch_size=32)
# Final evaluation of the model
scores = model.evaluate(X_test_rnn, y_test, verbose=0)
【问题讨论】:
标签: python neural-network lstm recurrent-neural-network