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)`