keras 的可视化

依赖安装

基础环境:

• keras.__version__ 2.0.8
• python sys.version 3.5.4
• windows 10

安装依赖

• GraphViz
  Graph Visualization Software, 下载地址. 记得添加C:\Program Files (x86)\Graphviz2.38\bin 到环境变量.

# 可以使用这个代码片来验证安装成功与否.
# 它画出的图像见 图1
from graphviz import Digraph

dot = Digraph(comment='The Round Table')

dot.node('A', 'King Arthur')
dot.node('B', 'Sir Bedevere the Wise')
dot.node('L', 'Sir Lancelot the Brave')
dot.edges(['AB', 'AL'])
dot.edge('B', 'L', constraint='false')

print(dot.source)
dot.render('test-output/round-table.gv', view=True)

图1 上述示例代码画出的图像

• pydot_ng
  pip install pydot_ng

然后重启IDE.

模型拓扑可视化

# 使用下面的简单方法, 在model.compile()之后调用, 即可绘图
from keras.utils import plot_model
plot_model(semantic_model, to_file='model.png',show_shapes='True')

figure 自己设计的TextCNN 的网络拓扑