如何通过张量流估计的每个时期来评估测试数据集

时间:2018-03-21 12:16:19

标签: tensorflow tensorboard tensorflow-datasets tensorflow-estimator

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

1 个答案:

答案 0 :(得分:2)

我建议使用如下循环:

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))