Keras中网络可视化模块安装配置

1、安装graphviz模块

1.1 命令行输入 pip install graphviz

1.2 安装graphviz软件。官网地址为http://www.graphviz.org/

官网中有解压版和安装版，推荐解压版，因为我使用解压版成功了，安装版好像出了点问题- -

解压版：

https://graphviz.gitlab.io/_pages/Download/windows/graphviz-2.38.zip

1.3 将安装目录中的graphviz-2.38\release\bin添加进Path环境变量

2、安装pydot模块

这里，别的博主建议python2安装pydot，python3安装pydot_ng，亲自试过后还是不行，就直接将两个都装了。

pip install pydot

pip install pydot_ng

3、测试

# -*- coding: utf-8 -*-

from keras.models import Sequential
from keras.layers.core import Dense
from keras.optimizers import SGD
from keras.utils.vis_utils import plot_model
def run():
    # 构建神经网络
    model = Sequential()
    model.add(Dense(4, input_dim=2, init='uniform',activation='relu'))
    model.add(Dense(2, init='uniform',activation='sigmoid'))
    sgd = SGD(lr=0.05, decay=1e-6, momentum=0.9, nesterov=True)
    model.compile(loss='binary_crossentropy', optimizer=sgd, metrics=['accuracy'])
    # 神经网络可视化
    plot_model(model, to_file='E:/model.png')

if __name__ == '__main__':
    run()

效果如下图：