tensorflow - Google Colab (tensorflow) 一个可视化工具

Question

我正在尝试在 Google Colab 上可视化我的神经网络架构。然而，由于某种原因，它没有产生任何视觉效果。你能告诉我为什么我的代码不能在 Colab 上运行吗？我不能使用像 python IDE 这样的东西。它必须是 Google Colab。

!pip3 install keras
!pip3 install ann_visualizer
!pip install graphviz

from keras.models import Sequential  
from keras.layers import Dense  
from ann_visualizer.visualize import ann_viz  

network = Sequential()  
#Hidden Layer#1  
network.add(Dense(units=2, activation='relu',  kernel_initializer='uniform', input_dim=2))  
#Output Layer  
network.add(Dense(units=1, activation='sigmoid', kernel_initializer='uniform'))  
ann_viz(network, title="")

标签： tensorflowkerasgoogle-colaboratorygraphvizgraph-visualization

Answer 1

一种解决方法是给 ann_viz 调用一个文件名，然后下载该文件。

ann_viz(network, view=True, title="test", filename="visualized")

现在，当您单击左侧的文件夹时，您将看到输出文件为 PDF，文件名为“visualized”。