Keras實踐:模型可視化

Keras實踐:模型可視化

安裝Graphviz

  官方網址爲:http://www.graphviz.org/。我使用的是mac系統,因此我分享一下我使用時遇到的坑。web

  Mac安裝時在終端中執行:bash

brew install graphviz

  若卡在Updating Homebrew....,須要取消brew的自動更新:app

  

 

安裝PyDot

  

 

 加載模型並生成結構圖

from keras.models import load_model
from keras.utils.vis_utils import plot_model
mymodel = load_model("/Users/apple/PycharmProjects/MyDjango/myweb/CNNModel.hdf5")
plot_model(emotion_classifier,'model.png')
相關文章
相關標籤/搜索