keras的內置函數keras.utils.vis_utils.plot_model可以直接將搭建的神經網絡用流程圖表示出來
該函數原來為keras.utils.visualize_util.plot,在2017年3月1日的更新中作了修改
visualize_util改為vis_utils;plot改為plot_model;
使用原函數名會報錯import error
使用該函數需要額外的函數庫和軟件,以下說明引自博客:
https://www.jianshu.com/p/3a97c0a43beb
1. (使用pip install)安裝pydot和graphviz兩個函數庫
2. 安裝graphviz軟件:
https://graphviz.gitlab.io/_pages/Download/Download_windows.html
3. 將路徑加入系統變量,從我的電腦-屬性-高級系統設置-環境變量 進入
(圖片為引用,根據安裝路徑調整)
4. 檢查graphviz是否安裝成功,進入windows命令行,輸入命令:dot -version,若成功顯示版本信息,則認為安裝和添加路徑成功
5. 引用函數需要以下語句
from keras.utils.vis_utils import plot_model
import os
os.environ["PATH"] += os.pathsep +'D:/Program Files (x86)/Graphviz2.38/bin/' # 后一段路徑改為自己安裝的路徑
6. 繪圖語句如下:
plot_model(model,to_file='name.png',show_shapes=True)
結果如圖所示: