输入 0 与层 lstm_12 不兼容：预期 ndim=3，发现 ndim=2答案

【问题标题】：Input 0 is incompatible with layer lstm_12: expected ndim=3, found ndim=2输入 0 与层 lstm_12 不兼容：预期 ndim=3，发现 ndim=2
【发布时间】：2020-06-12 02:42:42
【问题描述】：

我是 ML 新手，正在尝试制作 RNN LSTM 模型。

我想使用 GridSearchCV 优化超参数。我要优化的是每层选择的层数和节点数。

这是生成模型的代码：

from keras.wrappers.scikit_learn import KerasClassifier
from sklearn.model_selection import GridSearchCV
def create_model(layers,activation):
    model = Sequential()
    for i,node in enumerate(layers):
        if i == 0:
            model.add(LSTM(units=node, input_shape=(x_train.shape[1],1)))
            model.add(Activation(activation))
            model.add(Dropout(0.2))            
        else:
            model.add(LSTM(units=node, input_shape=(x_train.shape[1],1)))
            model.add(Activation(activation))
            model.add(Dropout(0.2)) 
    model.add(Dense(units=1)) 
    model.compile(optimizer='adam',loss='mean_squared_error',metrics=['accuracy'])
    return model

这里是变量

layers=[[40,40],[30,30],[30,30,30],[30,30,30,30],[30,30,30,30,30]]
activations =['sigmoid','relu']
batch_size = [32,50]
epochs = [50]

然后我用 gridsearchcv 把它包起来

param_grid = dict(layers=layers,activation=activations,batch_size=batch_size,epochs=epochs)
grid = GridSearchCV(estimator=model,param_grid=param_grid)

当我这样做时

grid_result = grid.fit(x_train,y_train,verbose=3)

我收到了这个错误

ValueError: Input 0 is incompatible with layer lstm_14: expected ndim=3, found ndim=2

我不知道会发生什么。我的 x_train 形状是 (13871, 60, 1)，y_train 形状是 (13871,)。预先感谢您，我们将非常感谢您的帮助！

谢谢！

菲尔

【问题讨论】：

标签： machine-learning lstm valueerror grid-search

【解决方案1】：

错误消息实际上很好地解释了这一点。 LSTM 需要形状的时间序列输入（batch_size、timesteps、features）。您的第一个输入 lstm 层似乎是正确的。但是，LSTM 的输出不是一个序列。随后的 LSTM 层将不会收到适当的输入。

你也可以通过设置参数使LSTM输出为一个序列

return_sequences=True

请注意，您可能必须在最后一层中将返回序列设置为 false，然后再进行密集或执行展平操作。

这有帮助吗？

PS：你的if... else，条件一模一样。您打算稍后更改吗？

【讨论】：

您好，谢谢 Spill 博士。我再次拿起了这个项目，你的建议奏效了。是的 if..else 实际上是一样的，我已经改变了它。谢谢！
很高兴它有帮助。也许您可以接受关闭帖子以供未来读者阅读的答案。