Skip to main content

TensorBoard PR曲线

tensorboard_pr_curve.py 示例演示了如何将ClearML集成到使用TensorFlow和TensorBoard的代码中。

示例脚本执行以下操作:

  • examples项目中创建一个名为tensorboard pr_curve的实验。
  • 创建三个类,R、G和B,并从正态分布中生成RGB空间内的颜色。每个随机颜色的真实标签与生成它的正态分布相关联。
  • 使用其他三个正态分布计算每种颜色属于该类的概率。
  • 使用这些概率生成PR曲线。
  • 使用tensorboard.plugins.pr_curve.summary为每个类创建摘要。
  • ClearML 自动捕获 TensorBoard 输出、TensorFlow 定义以及控制台输出。

图表

ClearML Web UI中,PR曲线摘要显示在实验页面的图表下。

  • 蓝色PR曲线

    image

  • 绿色PR曲线

    image

  • 红色PR曲线

    image

超参数

ClearML 自动记录 TensorFlow 定义。它们出现在 配置 > 超参数 > TF_DEFINE 中。

image

控制台

所有其他控制台输出显示在控制台中。

image