keras模型結構可視化

Question

我想查看keras模型，如this。我用K.get_session().graph和0x7f2a8b809400

得到

tensorflow.python.framework.ops.Graph，但我看到這個圖，並將其保存。我使用tensorflow後端

Answer 1

安裝tensorboard

導入它

from keras.callbacks import TensorBoard 

加載到一個變量

tbCallBack = TensorBoard(log_dir='Graph', 
         histogram_freq=10, 
         write_graph=True, 
         write_images=True) 

，然後使用該作爲訓練的回調：

model.fit(x, y, ... 
      callbacks=[tbCallBack]) 

確保你有一個名爲'Graph'的目錄或任何你想要的。然後前終端跑步訓練：

tensorboard --logdir Graph 

然後你就可以看到你的瀏覽器