站点图标 多伦多书苑

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

20.14 可视化神经网络

问题描述

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

解决方案

使用Keras的model_to_dot或者plot_model(见代码后的图):

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

讨论

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

Author:

退出移动版