如何檢查在python代碼中使用LSTM創建的順序模型？

Question

我有一個用python編寫的算法，它是使用LSTM進行時間序列分析。 我的教授要求我顯示代碼中創建的模型的詳細信息。 如何在這里檢查“模型”？ 它在后台具有模型的可視化效果嗎？

model = Sequential()
model.add(LSTM(50, input_shape=(trainX.shape[1], trainX.shape[2])))
model.add(Dropout(0.2))
model.add(Dense(1))
model.compile(loss='mae', optimizer='adam')
history = model.fit(trainX, trainY, epochs=50, batch_size=72, validation_data=(testX, testY), verbose=0, shuffle=False)

Answer 1

plot_model有一個可視化工具，稱為plot_model 。 您可以使用它將模型另存為圖像，在其中可以看到模型的結構，包括輸入和輸出尺寸。

from keras.utils import plot_model
plot_model(model, to_file='model.png')

您可以在這里閱讀有關它的更多信息： Keras Visualization