Keras模型可视化

python2

安装graphviz模块    sudo pip install graphviz

安装graphviz软件    sudo apt-get install graphviz

安装pydot模块    sudo pip install pydot==1.1.0

#只能装1.1.0版本，因为最新版（截止2016.8最新版本号是1.2.x）中find_graphviz函数是deprecated的，使用时会报错

这部分的顺序不能错！！！

python3

安装graphviz模块    sudo pip3 install graphviz

安装pydot模块更换为    sudo pip3 install pydot_ng 

测试方法

# encoding: utf-8
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.visualize_util import plot
# from keras.utils import plot_model


def run():
    # 构建神经网络
    model = Sequential()
    model.add(Dense(4, input_dim=2, init='uniform'))
    model.add(Activation('relu'))
    model.add(Dense(2, init='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, to_file='model.png')
    # plot_model(model, to_file='model.png',show_shapes='True')

if __name__ == '__main__':
    run()

输出的是简单版的网络结构图

plot_model(temp_model, to_file='model.png',show_shapes='True')可输出完全版的网络结构图