windos下keras可视化——graphviz&pydot_ng

1. 安装pydot_ng，graphviz

pip install pydot_ng
pip install graphviz

2. 计算机安装graphviz

下载链接：https://graphviz.gitlab.io/download/

并添加路径到环境变量，安装过程中有提示，或者自己添加"D:\Program Files\Graphviz 2.44.1\bin"到PATH

3.配置

修改pydot_ng的__init__.py文件中path的路径为Graphvie的实际bin目录路径

从他上面的解释说，默认去C:\Program Files 中尝试

我的安装路径是：D:\Program Files\Graphviz 2.44.1\bin

所以需要替换。

示例代码：

import numpy as np
from keras.models import Sequential
from keras.layers.core import Dense, Activation
from keras.optimizers import SGD
from keras.utils import np_utils
from keras.utils.vis_utils import plot_model
 
def run():
    # 构建神经网络
    model = Sequential()
    model.add(Dense(4, input_dim=2, kernel_initializer='uniform'))
    model.add(Activation('relu'))
    model.add(Dense(2, kernel_initializer='uniform'))
    model.add(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='model.png')
if __name__ == '__main__':
    run()