安妮 编译自 Google Research Blog
量子位 出品 | 公众号 QbitAI

今天，谷歌发布了一系列TensorBoard API，开发者可在TensorBoard中添加自定义的可视化插件，实现自定义可视化效果。同时，谷歌还升级了TensorBoard的仪表盘。

API获取地址：

https://github.com/tensorflow/tensorboard-plugin-example/blob/master/README.md

这些API有何特点？谷歌背后的目的何在？我们不妨一探究竟。

△ TensorBoard完全配置时的样子

由难到易

这事还得从头说起。

2015年，谷歌开源了TensorFlow，里面自带一套帮助检测、理解和运行模型的可视化工具TensorBoard。

TensorBoard中包含了一小组预先确定的通用可视化效果，基本上适用于所有的深度学习应用，比如观察损失随时间的变化，或在高维空间中探索聚类。

然而，在没有可重用的API时，TensorFlow团队外的开发人员添加新的可视化效果比较困难。因此，谷歌决定发布一套统一的API，让开发者能在TensorBoard中添加自定义的可视化插件。同时，谷歌还用这些新API升级了现有的仪表盘，当做创作参考示例。

举个栗子

目前，用户可以在GitHub上找到TensorBoard/plugins目录，探索这些TensorBoard中的插件列表。

目录地址：

https://github.com/tensorflow/tensorboard-plugin-example

举个例子，比如这个能生成准确率与召回率曲线的新插件pr_curves：

这个插件展示了标准TensorBoard插件中包含的三部分内容：

1. 用于收集数据供以后可视化使用的summary_operation文档

GitHub地址：

https://github.com/tensorflow/tensorboard/blob/master/tensorboard/plugins/pr_curve/summary.py

2. 为自定义数据服务的Python后端

GitHub地址：

https://github.com/tensorflow/tensorboard/blob/master/tensorboard/plugins/pr_curve/pr_curves_plugin.py

3. 内置TypeScript和polymer的TensorBoard仪表盘

GitHub地址：

https://github.com/tensorflow/tensorboard/tree/master/tensorboard/plugins/pr_curve/tf_pr_curve_dashboard

此外，与其他插件一样，“pr_curves”提供了一个Demo，用户可以学习如何使用插件，插件开发者可用来生成开发期间的示例数据。

为了进一步说明插件是如何工作的，谷歌还创建了一个框架性的Greeter插件，它能在运行模型时可以收集并显示问候语。谷歌在博客中建议开发人员从Greeter插件和其他现有的插件开始探索。