《Keras深度学习实战》—3.3.2　怎么做

3.3.2　怎么做

接下来，创建一个简单的模型并调用plot_model。

Keras中通过plot_model()函数将神经网络绘制成图形。函数包括以下参数：

• model（必需）：要绘制的模型

• to_file（必需）：保存模型图的文件名称

• show_shapes（可选，默认为False）：布尔值，用于显示每层的输出维度

• show_layer_names（可选，默认为True）：布尔值，用于显示每层的名称

以下部分介绍如何使用plot_model。

代码清单

创建一个包含两个Dense层的Sequential模型：

在Keras中可以使用summary()方法进行总结：

对该模型调用plot_model()函数：

输出图保存在以下文件中：