keras的模型可视化

keras提供了模型可视化的函数,但是要依赖一些工具,我们在win7系统安好Anaconda2和keras后,还要进行如下操作:

1、安装graphviz模块

1.1 命令行输入 pip install graphviz

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

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

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

2、安装pydot模块

命令行输入pip install pydot==1.1.0

以上两步的顺序一定要正确

做个实验看看吧:

#encoding=utf-8

from __future__ import absolute_import 
from __future__ import print_function

from keras.utils.visualize_util import plot
from keras.models import Sequential
from keras.layers.core import Dense, Dropout, Activation
from keras.layers.embeddings import Embedding
from keras.layers.recurrent import LSTM

model = Sequential()
model.add(Embedding(input_dim=1024, output_dim=256, input_length=50))
model.add(LSTM(128)) # try using a GRU instead, for fun
model.add(Dropout(0.5))
model.add(Dense(1))
model.add(Activation('sigmoid'))

plot(model, to_file='model1.png',show_shapes=True)

就是通过:

plot(model, to_file='model1.png',show_shapes=True)

生成一个模型图,第一个参数为模型,第二个为要生成图片的路径及文件名,还可以指定两个参数:
show_shapes:指定是否显示输出数据的形状,默认为False
show_layer_names:指定是否显示层名称,默认为True

看看效果:

show_shapes=false

keras的模型可视化_第1张图片

show_shapes=true

keras的模型可视化_第2张图片

第一个是show_shapes为false,不带输出形状的,第二个是带输出形状的,还不错!

你可能感兴趣的:(keras的模型可视化)