Tensorflow:保存评估指标

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可视化已保存的指标，方法如下：

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