參考: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
- pip install graphviz
- 到http://www.graphviz.org/Download_windows.php 中下載對應的msi並安裝
- 修改 D:***\python3.5\Lib\site-packages\pydot_ng__init__.py中的代碼,使pydot找到graphviz
可視化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’)
畫出的圖像如下: