tensorboard使用指南

tensorboard使用指南

代码部分(以一个最简单的CNN识别mnist手写数字集为例)：

首先要导入库

from keras.callbacks import TensorBoard

之后设置各种参数

tb = TensorBoard(log_dir=‘./log',  # log 目录
                 histogram_freq=1,  # 按照何等频率（epoch）来计算直方图，0为不计算
                 batch_size=64,     # 用多大量的数据计算直方图
                 write_graph=True,  # 是否存储网络结构图
                 write_grads=True,  # 是否可视化梯度直方图
                 write_images=True, # 是否可视化参数
                 embeddings_freq=0,
                 embeddings_layer_names=None,
                 embeddings_metadata=None)
callbacks = [tb]

fit部分的改动

model.fit(X_train, y_train, epochs=1, batch_size=64, validation_data=(X_test,y_test),
          callbacks=callbacks)

主要是添加validation_data和callbacks=callbacks，validation_data处应加入预测用的数据集和标签集

tensorboard的启动：

macOS或者Linux在终端输入

tensorboard —logdir=./log #等号右侧是之前保存的文件目录

Win需要找到tensorboard.exe所在的目录，在目录中打开cmd运行

.\tensorboard —logdir=.\log #等号右侧是之前保存的文件目录

Linux或者macOS下：

win下：

将cmd中显示的网址粘贴到浏览器中即可

如果网址粘贴到浏览器中无法进入，直接采用以下网址；

http://localhost:6006

显示效果：