Tensorflow:保存评估指标

时间:2018-09-03 16:31:29

标签: python 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

感谢您的帮助。

1 个答案:

答案 0 :(得分:1)

如果指定了model_dir,则度量标准将保存在此目录中。如果未明确指定,则将在/ tmp /中分配目录,并且该目录名称将打印在日志中。

您可以使用tensorboard通过以下方式可视化保存的指标:

Vastagság: 1,5