第一部分:如何实现可视化
第一种:利用 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
完成第二部分的“如何处理错误”,是不是第一部分的代码就不报错了?
完成!撒花🎉
网友评论