张量流:保存评估指标

Question

我以分布式方式训练了一个逻辑回归模型，使用 tensorflow 预制模型tf.Estimator.LogisticClassifier。 每 X 秒创建一个检查点，并对每个检查点执行一次评估。

是否可以保存每个检查点获得的评估指标？

我注意到评估指标可能已经保存，但我没有弄清楚可以保存它们的位置以及如何阅读它们：

INFO:tensorflow:Saving dict for global step 2000: accuracy = 1.0, accuracy_baseline = 1.0, auc = 1.0, auc_precision_recall = 0.0, average_loss = 0.0060327444, global_step = 2000, label/mean = 0.0, loss = 1.5422962, precision = 0.0, prediction/mean = 0.005973566, recall = 0.0

感谢您的帮助。

Answer 1

如果您指定了model_dir，则指标将保存在此目录中。如果未显式指定，则将分配/tmp/中的目录，并将此目录名称打印在日志中。

您可以使用张量板可视化保存的指标：

tensorboard --logdir /path/to/your/model_dir