Tensorflow估算器:如何在估算器的model_fn中获取估算名称?

时间:2020-04-14 20:40:11

标签: tensorflow tensorflow-estimator

在估算器中,我们可以在estimator.evaluate(..., name=(...))中或作为EvalSpec(..., name=(...))的一部分传递评价的名称。 Estimator创建一个默认的摘要编写器,并将摘要记录在eval_{name}目录中。这些摘要包括global_steps_per_secondloss

我使用两个不同的数据集,并分别使用不同的名称执行两次评估,并且在训练和评估期间都记录了一些张量。为了将这些张量(或任何文件)写入该目录,我需要model_fn()中的目录名称,因此如果mode为{{1},我可以使用它来获取实际的eval目录并适当地记录日志}。

估计符有tf.estimator.ModeKeys.PREDICT method,但是,正如您在source中看到的那样,当传递.eval_dir参数时,它只是返回eval_{name}

如何在估算器的name中获得name的评估?

0 个答案:

没有答案