Tensorflow將預測值轉換爲二進制

Question

我創建了一個應該將人分類爲超過50k或更少的神經網絡。當我輸出一個預測時，我得到的值如[2.06434059 -2.0643425]。但我需要他們在[1，0]或[0，1]。是否有任何可以轉換預測的張量函數？還是我必須手動執行？

在此先感謝

Answer 1

以你輸出的SOFTMAX。例如。

output = tf.nn.softmax(input) 

它會將值轉換爲概率分佈，其中的值將總和爲1。

否則，如果您只想{1,0}取最大值進來。