（给机器学习算法与Python学习加星标，提升AI技能）
ANN Visualizer 是一个很不错的 Python 库，兼容 Keras，它使用 Python 的 graphviz 库来创建开发人员正在构建的神经网络的可视化图形。

Github：
https://github.com/Prodicode/ann-visualizer
安装
From Github
  1. 从 Github 上下载 ann_visualizer 文件夹。
  2. 将 ann_visualizer 文件夹放在与主Python脚本相同的目录中。
From pip
使用以下命令：
pip install ann_visualizer
用法
from
ann_visualizer.visualize
import
ann_viz;

#Build your model here
ann_viz(model)
文件
ann_viz(model, view=True, filename="network.gv", title="MyNeural Network")
  • model - Keras Sequential 模型
  • view - 如果为 True，则在执行后打开图预览
  • filename - 保存图形的位置。（.gv 文件格式）
  • title - 图形的标题
示例
import
keras;

from
keras.models
import
Sequential;

from
keras.layers
import
Dense;


network = Sequential();

#Hidden Layer#1
network.add(Dense(units=
6
,

                 activation=
relu
,

                 kernel_initializer=
uniform
,

                 input_dim=
11
));


#Hidden Layer#2
network.add(Dense(units=
6
,

                 activation=
relu
,

                 kernel_initializer=
uniform
));


#Exit Layer
network.add(Dense(units=
1
,

                 activation=
sigmoid
,

                 kernel_initializer=
uniform
));


from
ann_visualizer.visualize
import
ann_viz;


ann_viz(network, title=
""
);
输出结果：
推荐阅读
【Github】100+ Chinese Word Vectors 上百种预训练中文词向量
手绘日漫版的新垣结衣见过没？这个开源动漫生成器让你的照片秒变
教程资源 | 5天玩转PyTorch深度学习，从GAN到词嵌入都有实例
技巧 | 30招教你顺滑切换Python3
继续阅读
阅读原文
关键词
模型
神经网络
文件夹
命令