用於可視化人工神經網路的 Python庫——ANN Visualizer

ANN Visualizer 是一個很不錯的 Python 庫，兼容 Keras，它使用 Python 的 graphviz 庫來創建開發人員正在構建的神經網路的可視化圖形。

Github：

https://github.com/Prodicode/ann-visualizer

安裝

From Github

從 Github 上下載ann_visualizer文件夾。

將ann_visualizer文件夾放在與主Python腳本相同的目錄中。

From pip

使用以下命令：

pipinstallann_visualizer

用法

fromann_visualizer.visualizeimportann_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- 圖形的標題

示例

importkeras;

fromkeras.modelsimportSequential;

fromkeras.layersimportDense;

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

fromann_visualizer.visualizeimportann_viz;

ann_viz(network, title="");

輸出結果：

計算機視覺基礎（從演算法到實戰應用）班

限時拼團，最後1天

已有100+人參加了此拼團

最高每人優惠200元！

新人福利

關注 AI 研習社（okweiwu），回復1領取

【超過 1000G 神經網路 / AI / 大數據資料】

Uber 開源「神經演化」可視化工具 VINE

喜歡這篇文章嗎？立刻分享出去讓更多人知道吧！