Keras plot_model問題解決辦法

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)
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章