如何在每个时期评估测试数据集的张量流估计器

Posted 2023-03-27

【中文标题】如何在每个时期评估测试数据集的张量流估计器

【英文标题】：How to evaluate testing dataset by every epoch for tensorflow estimator

【发布时间】：2018-08-30 13:30:29

【问题描述】：

对于 tf.estimator，它将运行所有 epoch，然后进行评估，但是如何按每个 epoch 评估测试数据集，以便我们可以在 tensorboard 中看到一条线而不是单个点

【参考方案1】：

我建议使用如下循环：

def input_fn(files):
  ds = dataset...
  return ds (without repeat)
for _ n_epochs:
  estimator.train(input_fn(train-files))
  estimator.evaluate(input_fn(eval-files))

【讨论】：

似乎会话将被创建和结束一次又一次。