1. 程式人生 > >tensorflow常用函式之tf.nn.softmax

tensorflow常用函式之tf.nn.softmax

關於softmax的詳細說明,請看Softmax。 
通過Softmax迴歸,將logistic的預測二分類的概率的問題推廣到了n分類的概率的問題。通過公式 
 
可以看出當月分類的個數變為2時,Softmax迴歸又退化為logistic迴歸問題。

 

下面的幾行程式碼說明一下用法

import tensorflow as tf
A = [1.0,2.0,3.0,4.0,5.0,6.0]
with tf.Session() as sess:
        print sess.run(tf.nn.softmax(A))

結果

[ 0.00426978  0.01160646  0.03154963  0.08576079  0.23312201  0.63369131]

 

轉自TensorF學習網 :https://my.oschina.net/u/780234/blog/1588827