Keras 可視化 model


參考:https://keras.io/visualization/
error解決參考:http://blog.csdn.net/wangjian1204/article/details/50346457
平台: win7 Python3.5

安裝附加依賴項

  • 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
    1. pip install graphviz
    2. http://www.graphviz.org/Download_windows.php 中下載對應的msi並安裝
    3. 修改 D:***\python3.5\Lib\site-packages\pydot_ng__init__.py中的代碼,使pydot找到graphviz
    • 注釋掉下面的if中的部分
    • Method 2 不做修改
    • Method3 添加
      path = r”D:\software\graphviz\bin”
      該路徑指向剛才下載的msi的安裝路徑中的bin

可視化model的方法

from keras.utils.visualize_util import plot

answer = Sequential()
answer.add(Merge([response, question_encoder], mode=’concat’, concat_axis=-1))
answer.add(LSTM(32))
answer.add(Dropout(0.3))
answer.add(Dense(vocab_size))
answer.add(Activation(‘softmax’))

plot(answer, to_file=’answer_model.png’)

畫出的圖像如下:






免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM