python关于如何进行Keras神经网络可视化的配置

来源:互联网 发布:淘宝如何提升访客量 编辑:程序博客网 时间:2024/06/07 12:22

python关于如何进行Keras神经网络可视化的配置

keras是一款非常不错的深度学习框架,并且它自身也带有神经网络可视化的函数plot_model(笔者使用的是2.08较早的版本为plot)。

但是,使用plot_model函数需要依赖于graphviz 和pydot。使用pip install安装即可。

两款依赖包安装好以后其实笔者还是遇到了问题:
pydot.InvocationException: GraphViz’s executables not found
就是 找不到graphviz()函数。
打开C:\anacondo\Lib\site-packages\pydot_ng\__init__.py
在find_graphviz()进行修改,这里我参考了

http://blog.csdn.net/shouwangzhelv/article/details/51163535
中的方法。(该文中是对pydot.py进行修改,目前最新版本的pydot
版本对__init__.py 进行修改)

from __future__ import absolute_import from __future__ import print_functionfrom keras.utils.vis_utils import plot_modelfrom keras.models import Sequentialfrom keras.layers import Dense, Activationmodel = Sequential() #就是多个网络层的线性堆叠model.add(Dense(32, batch_input_shape=(None, 784)))plot_model(model, to_file='model1.png',show_shapes=True)`

这里写图片描述

阅读全文
0 0
原创粉丝点击