API TensorFlow Estimator: использовать функции numpy? - PullRequest
0 голосов
/ 05 декабря 2018

Я написал несколько пользовательских метрик, которые я хотел бы добавить к 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)
    ...
...