如何安装keras模型可视化工具(Visualization utils)

目录

  • 写在前面
  • 安装步骤:
  • 测试代码
  • Reference

写在前面

之前安装pydot实在是踩了许多的坑,当时就记录一下安装笔记,今天实验室师弟遇到了同样的问题,我就把之前整理的笔记发给了他,在CSDN也备份一下。后面遇到问题了能够快速解决。
首先来说keras可视化的意义,那我想应该就会是两个方面:
1、学习别人的模型,有时候读代码可能会很费劲,可是一 看到流程图就都明白了,因此方便学习。
2、就是可以通过对自己代码的分析,从而可以更方便 的调试并改进模型。

安装步骤:

windows 10 keras2.08.
1、命令行输入 pip install graphviz
2、安装graphviz软件。官网地址为http://www.graphviz.org/
解压版:配置环境变量。将安装目录中的graphviz-2.38\release\bin添加进Path环境变量
安装版:安装msi
3、命令行输入pip install pydot==1.1.0
注:此处需要指定安装1.1.0版本的pydot,是因为最新版(截止2016.8最新版本号是1.2.x)中find_graphviz函数是deprecated的,使用时会报错1

注意事项:
1、一定不要把顺序弄反,按照1——2——3的顺序进行安装,否则最后pydot是无法找到graphviz的。
2、要注意的是安装完graphviz之后要将系统路径:C:\Program Files (x86)\Graphviz2.38\bin 添加到path环境变量中。然后再进行安装pydot
3、对于python3.x来说安装pydot-ng win下安装指令 pip install pydot-ng2

另外一点就是,现在的可视化模块不叫作:keras.utils.visualize_util 了,改成了keras.utils.vis_utils了,因此导入的时候要特别的注意。等所有的错误都不报的时候,我们就完成了绘图的过程:

测试代码

下面是测试代码3

from keras.models import Sequential
from keras.layers import LSTM, Dense
#这一行新加的,用于导入绘图包
from keras.utils.vis_utils import plot_model

import numpy as np
data_dim = 16
timesteps = 8
num_classes = 10

# expected input data shape: (batch_size, timesteps, data_dim)
model = Sequential()
model.add(LSTM(32, return_sequences=True,
               input_shape=(timesteps, data_dim)))  # returns a sequence of vectors of dimension 32
model.add(LSTM(32, return_sequences=True))  # returns a sequence of vectors of dimension 32
model.add(LSTM(32))  # return a single vector of dimension 32
model.add(Dense(10, activation='softmax'))

model.compile(loss='categorical_crossentropy',
              optimizer='sgd',
              metrics=['accuracy'])

# Generate dummy training data
x_train = np.random.random((1000, timesteps, data_dim))
y_train = np.random.random((1000, num_classes))

# Generate dummy validation data
x_val = np.random.random((100, timesteps, data_dim))
y_val = np.random.random((100, num_classes))

model.fit(x_train, y_train,
          batch_size=64, epochs=1,
          validation_data=(x_val, y_val))
#这一行新加的,用于绘图
plot_model(model, to_file='model1.png',show_shapes=True)

Reference


  1. python的数据可视化 graphviz pydot安装配置(win10) ↩︎

  2. Keras中神经网络可视化模块keras.utils.visualize_util 的安装配置与错误解决办法 ↩︎

  3. 入手Keras的一系列问题(1) ↩︎

你可能感兴趣的:(pyLearning,深度学习)