首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >如何检查在python代码中使用LSTM创建的顺序模型?

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

Stack Overflow用户
提问于 2018-04-19 08:41:18
回答 1查看 69关注 0票数 0

我有一个用python编写的算法,它是使用LSTM进行时序分析的。我的教授要求我展示在代码中创建的模型的细节。我怎么检查这里的“模型”?它是否在背景中对模型进行了可视化?

代码语言:javascript
运行
复制
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)
EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2018-04-19 09:09:01

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

代码语言:javascript
运行
复制
from keras.utils import plot_model
plot_model(model, to_file='model.png')

你可以在这里读到更多关于它的信息:Keras可视化

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/49916517

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档