《Gluon 動手學深度學習 六》Softmax實現

線性迴歸適用於輸出爲連續值的情景,對於輸出離散值的分類問題,需要使用分類模型,例如softmax,與線性迴歸不同的是softmax輸出單元從一個變爲多個。和線性迴歸相同的是,softmax也是一個單層神經網絡。和線性迴歸不同的是,softmax輸出層的個數等於類別個數。 softmax 每個輸出的計算都要依賴於上一層的每個元素,所以softmax輸出層是一個全連接層。 舉例:圖片分類問題 假設輸入
相關文章
相關標籤/搜索