Tensorflowの各クラスの生の確率値を取得する方法は？

Question

N個のクラスのそれぞれに対して、テストされたサンプルの類似性の生の確率値を得ることができます（私の使用例はTensorFlow Mechanics 101チュートリアルに基づいています）。

たとえば、tf.nn.in_top_k(logits, labels, 1)を実行する代わりに、テストされたサンプルが> = 0.25の確率で2番目のラベルに似ている場合は、Trueを返すしきい値を設定します。

Answer 1

greater_equalメソッドを使用できます。例えば

threshold = tf.constant(0.25, dtype=tf.float32) 
raw_prob = tf.greater_equal(predictions, threshold) 

raw_probはTrue又はFalse保持predictionsと同じサイズのテンソルであろう。異なるクラスに異なるしきい値を使用することもできます。クラス数が3であると仮定すると、

threshold = tf.constant([0.34, 0.25, 0.95], dtype=tf.float32) 
raw_prob = tf.greater_equal(predictions, threshold)