Python机器学习手册：从数据预处理到深度学习

问题描述

快速可视化神经网络的结构。

解决方案

使用Keras的model_to_dot或者plot_model（见代码后的图）：

如果想把上面的网络结构图保存为文件，可以使用plot_model：

讨论

Keras提供了工具函数用于快速可视化神经网络。如果想在Jupyter Notebook中显示一个神经网络，可以使用model_to_dot。show_shapes参数指定是否展示输入和输出的形状，它可以帮助我们调试网络。如果想展示一个更简单的模型，可以设置show_shapes=False（见代码后的图）：