文档平台

外观

在 keras model 生成 tensorboard graph 图 ​

  1. 导入 tensorboard

    from keras.callbacks import TensorBoard

  2. 在 Model 的 fit 函数内加 callback 参数

    mode.fit(..., callbacks = [TensorBoard(log_dir="/path/dir")])

这样生成的 TensorBoard 文件在 /path/dir 下：

在 存在 tensorboard 命令的系统输入：

tensorboard --logdir=/path/dir

即可打开 http://host:6006/ 查看 Graph

如果在 Linux、OSX 等系统没有 tensorboard 命令，可输入：

python /Library/Frameworks/Python.framework/Versions/3.5/lib/python3.5/site-packages/tensorboard/main.py --logdir=/path/dir

可写入一个 shell 文件，放到 /usr/local/bin/tensorboard

vim /usr/local/bin/tensorboard
#!/usr/bin/env bash
python /Library/Frameworks/Python.framework/Versions/3.5/lib/python3.5/site-packages/tensorboard/main.py $@
chmod +x /usr/local/bin/tensorboard

这样即可在任何路径使用 tensorboard