自动将类似 Tensorboard 的损失图保存到图像文件中

Str*_*che 5 tensorflow tensorflow-estimator

我目前正在尝试使用 tensorflow 预制估计器(tf.estimator.DNNRegressor)来预测粒子的运动。

我想将平均损失图的图像(如一个张量板显示)保存到每个模型的文件夹中。来自张量板的图像

Tensorboard 非常适合在训练期间监控这一点,但我想保存图像以供将来参考(例如,直观地比较不同的方法)

是否有捷径可寻?我可以在不同时间保存评估结果并使用 matplotlib,但是我没有找到任何关于如何从 regressor.train 方法中获取损失的信息。

小智 6

是的,目前,您可以下载@Jdehesa在评论中指出.svg的 loss/other_metric 历史记录:

  1. 检查Show data download links;

  2. 您可以在图的底部看到一个下载符号;

  3. 如果找不到,请参见图中的黄色标记。

图片