keras模型结构可视化

Question

我想看像这这样的keras模型。我用了K.get_session().graph

tensorflow.python.framework.ops.Graph 0x7f2a8b809400

但我要看这张图并保存它。我正在使用tensorflow后端

Answer 1

安装张紧板

进口它

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])

确保您有一个已生成的目录，名为“图形”或任何您想要的。然后在航站楼进行训练之前：

tensorboard --logdir Graph

然后您可以在浏览器中看到您的图形。