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

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_function

from keras.utils.vis_utils import plot_model
from keras.models import Sequential
from keras.layers import Dense, Activation

model = Sequential() #就是多个网络层的线性堆叠
model.add(Dense(32, batch_input_shape=(None, 784)))
plot_model(model, to_file='model1.png',show_shapes=True)`