from keras.utils.vis_utils import plot_model
直接調用keras中的可視化函數會報錯,原因是keras源代碼中出現了問題,我用的操作系統是window 10,下面的操作可以解決plot_model的問題。。
安裝附加依賴項
pydot
pip install pydot_ng
官方文檔中說直接安裝pydot,但是由於keras中使用了pydot.find_graphviz()函數,而這一函數在1.2.*後的版本被剝離了,所以不能直接pip安裝pydot,而是安裝pydot_ng.
在keras的visualize_util.py中,import pydot或pydot_ng的源碼如下:
graphviz
• pip install graphviz
• 到http://www.graphviz.org/Download_windows.php 中下載對應的msi並安裝
• 修改 D:*\python3.5\Lib\site-packages\pydot_ng__init__.py中的代碼,使pydot找到graphviz
o 註釋掉 下面的if中的部分
o Method 2 不做修改
o Method3 添加
path = r”D:\software\graphviz\bin”
該路徑指向剛纔下載的msi的安裝路徑中的bin
源代碼修改好後,可以檢驗下:
from keras.utils.vis_utils import plot_model
plot_model(model,to_file = filepath + filename,show_shapes = True)