keras模型结构可视化

Question

我希望看到像this这样的keras模型。我使用了K.get_session().graph并获得了

tensorflow.python.framework.ops.Graph at 0x7f2a8b809400

但我要看这个图并保存。我正在使用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

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