环境:win10+python3+keras2
使用keras的打印模型结构图
keras.utils.plot_model(model, to_file='model.png')
会遇到如下错误
RuntimeError: Failed to import pydot. You must install pydot and graphviz for pydotprint to work
使用pip安装上面提示的2个包后,依然保存。网友说这个是graphviz的问题,需完全安装graphviz(graphviz下载地址)
安装后记得添加graphviz/bin到系统环境变量
测试:
mport 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 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()
效果图: