tf.keras.callbacks.TensorBoard

概述

用于在keras中实现 TensorBoard

官方网址:
https://www.tensorflow.org/api_docs/python/tf/keras/callbacks/TensorBoard

主要方法

初始化

__init__():
    log_dir='logs',
    histogram_freq=0,
    write_graph=True,
    write_images=False,
    update_freq='epoch',
    profile_batch=2,
    embeddings_freq=0,
    embeddings_metadata=None,
    **kwargs
参数
  • log_dir: 保存日志文件的路径+文件名。

  • histogram_freq: 对于模型中各个层计算激活值和模型权重直方图的频率(训练轮数中)。必须设置此参数
    如果设置成 0 ,TensorBoard的distributions和histograms仪表盘不显示。

  • write_graph: 是否在 TensorBoard 中可视化图像。 如果 write_graph 被设置为 True,日志文件会变得非常大。

  • write_grads: 是否在 TensorBoard 中可视化梯度值直方图。 histogram_freq 必须要大于 0 。

  • batch_size: 用以直方图计算的传入神经元网络输入批的大小。

  • write_images: 是否在 TensorBoard 中将模型权重以图片可视化。

  • embeddings_freq: 被选中的嵌入层会被保存的频率(在训练轮中)。

  • embeddings_layer_names: 一个列表,会被监测层的名字。 如果是 None 或空列表,那么所有的嵌入层都会被监测。

  • embeddings_metadata: 一个字典,对应层的名字到保存有这个嵌入层元数据文件的名字。 查看 详情 关于元数据的数据格式。 以防同样的元数据被用于所用的嵌入层,字符串可以被传入。

  • embeddings_data: 要嵌入在 embeddings_layer_names 指定的层的数据。 Numpy 数组(如果模型有单个输入)或 Numpy 数组列表(如果模型有多个输入)。 Learn ore about embeddings。

  • update_freq: ‘batch’ 或 ‘epoch’ 或 整数。当使用 ‘batch’ 时,在每个 batch 之后将损失和评估值写入到 TensorBoard 中。同样的情况应用到 ‘epoch’ 中。如果使用整数,例如 10000,这个回调会在每 10000 个样本之后将损失和评估值写入到 TensorBoard 中。注意,频繁地写入到 TensorBoard 会减缓你的训练。

感谢

https://blog.csdn.net/zhangpeterx/article/details/90762586


原文出处:/show_1EF4D4mLap8Z.html