macOS下实现Keras模型的可视化及错误处理
2018-02-17 本文已影响124人
之城之城
第一部分：如何实现可视化
第一种：利用 plot_model 导出模型结构图
keras.utils.vis_utils模块提供了画出Keras模型的函数，依赖于graphviz。
该函数将已经设计好的模型画成结构图，并保存成图片：
from keras.utils import plot_model
plot_model(model, to_file='model.png')
导出的模型图
plot_model接收两个可选参数：
show_shapes：指定是否显示输出数据的形状，默认为False
show_layer_names:指定是否显示层名称,默认为True
第二种：在ipython直接展示模型结构图
我们也可以直接获取一个pydot.Graph对象，然后按照自己的需要配置它，例如，如果要在ipython中展示图片
from IPython.display import SVG
from keras.utils.vis_utils import model_to_dot
SVG(model_to_dot(model).create(prog='dot', format='svg'))
如下图所示：
在ipython直接展示的模型结构图
如何处理错误
由于上述代码依赖 pydot 和 graphviz，未安装就会出现错误。
- 安装 graphviz
第一步：获取权限
sudo chown -R `whoami`:admin /usr/local/bin
sudo chown -R `whoami`:admin /usr/local/share
第二步：安装
brew install graphviz
brew link --overwrite graphviz
- 安装pydot
pip install pydot
完成第二部分的“如何处理错误”，是不是第一部分的代码就不报错了？
完成！撒花🎉
