Google Colab (tenorflow) и визуализатор

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

Answer 1

Один из обходных путей - позвонить ann_viz по имени файла и затем загрузить файл.

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

Теперь, когда вы щелкнете папку с файлом слева, вы увидите выходной файл как PDF с именем файла «визуализированный».