Question

我有一个用python编写的算法，它是使用LSTM进行的时间序列分析。我的教授让我展示代码中创建的模型的细节。我如何检查＆＃34;模型＆＃34;这里？它是否在后台有一些模型可视化？

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

Keras中有一个名为plot_model的可视化工具。您可以使用它将模型保存为图像，您可以在其中查看模型的结构，包括输入和输出尺寸。

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

您可以在此处详细了解：Keras Visualization

如何检查在python代码中使用LSTM创建的顺序模型？

1 个答案: