TensorBoard（1）开始使用

一、TensorBoard作用

它能跟踪模型指标，例如损失和准确性，可视化模型图，将嵌入物投影到较低维度的空间等等。

二、通过Keras Model.fit()使用TensorBoard

2.1 TensorBoard回调函数

当使用Keras Model.fit() 函数训练时，添加tf.keras.callback.TensorBoard回调，可确保创建和存储日志。

在每个时期启用histogram_freq = 的直方图计算功能（默认情况下处于关闭状态）

将日志放在带有时间戳的子目录中，以便轻松选择不同的训练运行。

model = create_model()
model.compile(optimizer='adam',
              loss='sparse_categorical_crossentropy',
              metrics=['accuracy'])

log_dir="logs/fit/" + datetime.datetime.now().strftime("%Y%m%d-%H%M%S")
tensorboard_callback = tf.keras.callbacks.TensorBoard(log_dir=log_dir, histogram_freq=1)

model.fit(x=x_train, 
          y=y_train, 
          epochs=5, 
          validation_data=(x_test, y_test), 
          callbacks=[tensorboard_callback])

其中关键代码是 callbacks=[tensorboard_callback]) ，它在model.fit()中的回调接口指定了tensorboard_callback，这样训练过程中会存储日志，后面可以在TensorBoard中可视化查看。

2.2 打开TensorBoard

通过命令行终端 或在notebook中启动TensorBoard。

在notebook中，使用%tensorboard 命令：

%tensorboard --logdir logs/fit

在命令行中，运行是不带"%"的，命令基本一样：

tensorboard --logdir logs/fit

打开后如下图所示：

2.3 TensorBoard界面说明

Scalars 显示损失和指标在每个时期如何变化；还可以使用它来跟踪训练速度，学习率和其他标量值。

Graphs 帮忙我们可视化模型。它会显示Keras的结构图，便于了解层与层之间的连接关系。

Diatributions 和Histograms 显示张量随时间的分布。这对于可视化权重和偏差并验证它们是否可以预测的方式变化很有用。

Diatributions 界面如下：

Histograms 界面如下：

小结：

Scalars 显示损失和指标在每个时期如何变化；还可以使用它来跟踪训练速度，学习率和其他标量值。

Graphs 帮忙我们可视化模型。它会显示Keras的结构图，便于了解层与层之间的连接关系。

Diatributions 和Histograms 显示张量随时间的分布。这对于可视化权重和偏差并验证它们是否可以预测的方式变化很有用。

# Load the TensorBoard notebook extension.
%load_ext tensorboard

​

三、源代码

首先加载tensorboard

# Load the TensorBoard notebook extension.
%load_ext tensorboard

清楚之前的日志（可选）

# Clear any logs from previous runs
rm -rf ./logs/

再执行如下代码：

然后打开tensorboard即可。

在notebook中，使用%tensorboard 命令：%tensorboard --logdir logs/fit

在命令行中，运行是不带"%"的，命令基本一样：tensorboard --logdir logs/fit