python - Keras 中的 LSTM 和 fit_generator 错误 "You must compile your model before using it"

Question

我创建了自己的类，它在其中一个方法中创建了一个 Keras 模型。

self.model = Sequential()
self.model.add(LSTM(32))
self.model.add(Dense(2, activation='relu'))
self.model.compile(optimizer='RMSprop', loss='categorical_crossentropy', metrics=['acc'])

在其他方法中，我尝试使用 python 生成器作为数据提供者来训练这个模型。

self.model.fit_generator(my_gen(), steps=10, epochs=1, verbose=1)

这会导致错误:

raise RuntimeError('You must compile your model before using it.')
RuntimeError: You must compile your model before using it.

如果我将 LSTM 层更改为密集层，错误不会增加。我做错了什么？

带有 Tensorflow 1.8.0 后端的 Keras 2.2.0 版。

Answer 1

似乎第一个 Keras LSTM 层在使用 fit_generator 时仍然需要 input_shape，这在 Keras 文档中似乎缺失并导致“你必须编译你的模型在使用它之前”错误。

要解决这个问题，请确保您的第一个 LSTM 层中有一个 input_shape 参数，如下例所示:

model.add(LSTM(100, input_shape=(n_timesteps, n_dimensions), return_sequences=True))
model.add(Dropout(0.2))
model.add(LSTM(100, return_sequences=False))
model.add(Dropout(0.2))
model.add(Dense(10, activation='tanh'))

model.compile(loss='mse', optimizer='adam')