如何在Tensorflow中获取每个类的原始概率值?

时间:2016-08-11 09:50:29

标签: classification tensorflow

我如何获得测试样本与N类中每一类的相似性的原始概率值(我的用例基于TensorFlow Mechanics 101教程)?

例如,我想设置一个返回tf.nn.in_top_k(logits, labels, 1)的阈值,以防测试样本类似于第二个标签,概率为> = 0.25。

1 个答案:

答案 0 :(得分:1)

您可以使用greater_equal方法。例如

raw_prob

predictions的张量与TrueFalse threshold = tf.constant([0.34, 0.25, 0.95], dtype=tf.float32) raw_prob = tf.greater_equal(predictions, threshold) 相同。也可以对不同的类使用不同的阈值。假设课程数量是三个:

  <?php
  $abc ="/?goto=%2Fr%2Faccount%2Findex%2Ecfm%3Fsite_id%3D87211";

  $test = parse_url($abc);
  $test = urldecode($test["query"]);

  $url = parse_url($test);    
  parse_str($url['query'], $param);
  print_r($param["site_id"]);

  ?>