keras模型结构可视化
问题描述:
我想查看keras模型,如this。我用K.get_session().graph
和0x7f2a8b809400keras模型结构可视化
得到
tensorflow.python.framework.ops.Graph,但我看到这个图,并将其保存。我使用tensorflow后端
答
安装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
然后你就可以看到你的浏览器
什么是在上面的回答目录意味着你的图表。我正在使用当前的工作目录。我得到但不是图 –
Hitesh
你必须创建一个名为'Graph'的目录或任何你想在你的工作目录中,然后用它作为log_dir的参数,因此tensorflow知道哪个目录写日志到 – convolutionBoy
我遵循以上建议,并获得一些events.out.tfevents.1506331517文件,但再次不是图 – Hitesh