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 

然后你就可以看到你的浏览器