温馨提示×

Keras中如何可视化模型结构

小樊
97
2024-03-11 11:54:25
栏目: 深度学习
开发者测试专用服务器限时活动,0元免费领,库存有限,领完即止! 点击查看>>

要可视化Keras模型的结构,可以使用keras.utils.vis_utils模块中的plot_model函数。该函数可以将模型的结构图保存为图片或者显示在Jupyter Notebook中。

下面是一个示例代码,用于可视化一个简单的Keras模型的结构:

from keras.models import Sequential
from keras.layers import Dense
from keras.utils.vis_utils import plot_model

# 创建一个简单的Sequential模型
model = Sequential()
model.add(Dense(32, input_shape=(784,), activation='relu'))
model.add(Dense(10, activation='softmax'))

# 可视化模型结构并保存为图片
plot_model(model, to_file='model.png', show_shapes=True)

# 在Jupyter Notebook中显示模型结构
plot_model(model, show_shapes=True)

运行以上代码,会生成一个名为model.png的图片文件,显示了模型的结构图;同时在Jupyter Notebook中显示模型结构图。您可以根据实际需求调整模型结构,并根据需要调整plot_model函数中的参数。

亿速云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>

推荐阅读:Keras如何可视化模型结构

0