Tensorflowプロットはeval_metric_ops

Question

tf.metrics.precision_at_thresholds()介しTensorboardにtf.metrics.precision_at_thresholds 3つの引数を取る：labels, predictions, thresholds閾値は[0,1]との間の閾値のPythonリストまたはタプルあります。この関数は、eval_metric_opsを自動的にテンターボードにプロットするのに問題のある "float Tensor of shape [len（thresholds）]"を返します（私はスカラーであると信じています）。値はコンソールにはうまく印刷されますが、テンソルボードに値をプロットしたいと思います。テンソルボードで値をプロットできるように調整できることはありますか？

Answer 1

私の現在のアプローチは、リスト内の最初の要素の平均値だけを取る別の関数を作成することです。しかし、これよりも洗練されたソリューションがあると期待しています。

def metric_fn(labels, predictions, threshold): 
    precision, precision_op = tf.metrics.precision_at_thresholds(labels = labels, 
                predictions = predictions, 
                thresholds = threshold) 
    mean, op = tf.metrics.mean(precision[0]) 

    return mean, op