Skip to content

Blog

[Machine Learning] softmax 函式介紹與程式實作

Softmax

Softmax 函式,又被稱為『歸一化指數函數』,基本上是將一組向量(就好比說我們 Machine Learning 最後輸出的預測結果有多個分類,每個分類有著一個分數)映射為每個向量當中的元素都位於 (0, 1) 之間,其實就是代表著每個分類的機率分佈。當然,既然是機率分佈,那麼這個向量的所有元素相加總和應為 1。

Read More »[Machine Learning] softmax 函式介紹與程式實作

[PyTorch] Getting Start: Neural Networks 神經網路的基本介紹

所謂的『神經網路』,便是我們想要進行深度學習所建的『模型架構』。在 PyTorch 官方的教學裡頭,第一句話就明確地闡明了:

可以使用 torch.nn 來建構神經網路。

nn 包含了模型層以及一個 forward() 的函式,並且會返回 output。這在之後的程式碼中可以清楚看見。

Read More »[PyTorch] Getting Start: Neural Networks 神經網路的基本介紹