softmax這個函數(shù)用于多分類任務(wù)擎颖,在我們使用中經(jīng)常和交叉熵聯(lián)合起來蜀肘,至于交叉熵計算loss酌予,以后看到了會寫磺箕,
先放圖
softmax函數(shù)圖形如下
沒圖
這個函數(shù)不同于sigmoid函數(shù),函數(shù)圖形不好畫
放公式
這個
這個公式看上去可能有復(fù)雜抛虫,實際上wiki解釋就是松靡,將一組n維向量轉(zhuǎn)化為0,1之間的另外一組向量建椰。
這個函數(shù)同sigmoid函數(shù)一樣適用于分類任務(wù)击困,將線性回歸后tensor輸入到softmax,會得到一個對應(yīng)分類的概率广凸,
根據(jù)所給出概率實現(xiàn)分類任務(wù)阅茶,不同于sigmoid的二分類,softmax對任意的維度都適用,至于分類任務(wù)太多谅海,效果好不好我也沒有實踐過脸哀。好了,簡單介紹了softmax扭吁,有興趣的還可以深入
歡迎大家留言討論