TensorFlow Estimator API:使用numpy函数吗?

时间:2018-12-05 14:39:36

标签: python tensorflow tensorflow-estimator

我编写了一些自定义指标,我想将这些指标添加到tf.summary并通过metrics_op中的EstimatorSpec在张量板中查看。

是否严格要求我将这些函数隐式转换为它们的tf图等效项,或者是否有办法例如将其转储到其中?

def model_fn(...):

    ...
    my_scalar_metric = my_np_func(output, labels)
    tf.summary.scalar('my_metric', my_scalar_metric)
    ...

0 个答案:

没有答案