版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/zyxhangiian123456789/article/details/87368498
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()
效果如下图: