1.keras模型可視化 keras.utils.vis_utils模塊提供了畫出Keras模型的函數(利用graphviz) 該函數將畫出模型結構圖,並保存成圖片: plot_model接收兩個可選參數: show_shapes:指定是否顯示輸出數據的形狀,預設為False show_layer_ ...
keras.utils.vis_utils
模塊提供了畫出Keras模型的函數(利用graphviz)
該函數將畫出模型結構圖,並保存成圖片:
from keras.utils import plot_model plot_model(model, to_file='model.png')
plot_model
接收兩個可選參數:
show_shapes
:指定是否顯示輸出數據的形狀,預設為False
show_layer_names
:指定是否顯示層名稱,預設為True
2.需要安裝pydot-ng 和 graphviz依賴庫
分別sudo pip install之後還是會報錯 Failed to import pydot. You must install pydot and graphviz for `pydotprint` to work,可以通過 sudo apt-get install graphviz 解決.