Github 项目推荐 | 用于可视化人工神经网络的 Python 库 —— ANN Visualizer

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="");

输出结果：

计算机视觉基础（从算法到实战应用）班

限时拼团，最后1天

已有100+人参加了此拼团

最高每人优惠200元！

新人福利

关注 AI 研习社（okweiwu），回复  1  领取

【超过 1000G 神经网络 / AI / 大数据资料】

Uber 开源「神经演化」可视化工具 VINE

▼▼▼