大家好

最近在搞深度學習
發現很重要的是模型的架構
模型的架構很難說隨便選一種就能夠適應所有種類資料
很多還是有 雙input的模型
所以如何將模型架構顯示出來
討論並且修改
這也是一個重點
 
如果使用 summary() 產出會是這樣

如果有分支就不太容易看
 
 
所以用 plot_model  會比較方便
我這邊整理出重點是
 
下載 graphviz
 
當下位置 (有一定機率失效)
 
解壓縮到電腦中
C:\graphviz-2.44.1-win32/
 
安裝 pydot-ng 也就是 pip install pydot-ng
 
然後使用
 
from keras.utils import plot_model
os.environ["PATH"] += os.pathsep + 'C:\graphviz-2.44.1-win32/Graphviz/bin/'  # 安装graphviz的路径
 
plot_model(
    net, #你的模型
    to_file='model_n1.png', #要存放的圖片
    show_shapes=True,#顯示input output 維度
    show_layer_names=True #顯示層名稱
    )
 
就可以產生以下圖片

 
給大家參考囉!