ANN Visualizer 是一个很不错的 Python 库,兼容 Keras,它使用 Python 的 graphviz 库来创建开发人员正在构建的神经网络的可视化图形。
Github:
https://github.com/Prodicode/ann-visualizer
安装
From Github
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")
示例
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="");
输出结果: