lstm模型画图python代码

以下是使用Python和Keras绘制LSTM模型的示例代码：

from keras.models import Sequential
from keras.layers import LSTM, Dense
import matplotlib.pyplot as plt
from keras.utils.vis_utils import plot_model

# Define LSTM model
model = Sequential()
model.add(LSTM(64, input_shape=(10, 1)))
model.add(Dense(1))

# Compile the model
model.compile(loss='mean_squared_error', optimizer='adam')

# Plot the model
plot_model(model, to_file='lstm_model.png', show_shapes=True)

# Show the plot
img = plt.imread('lstm_model.png')
plt.imshow(img)
plt.show()

该代码使用Keras库创建一个包含一个LSTM层和一个密集层的模型，并使用`plot_model`函数将模型图形绘制到磁盘上的文件中。最后，使用`imshow`函数显示图像。你可以根据自己的需要修改模型的层数和大小，并更改保存的文件名和图像显示方式。