Skip to content

10 月 2019

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

Last Updated on 2021-05-02 by Clay

Softmax

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

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

[PyTorch] Getting Start: 訓練 CIFAR-10 資料集的分類模型

Last Updated on 2021-04-24 by Clay

今天再次挑戰了不同的資料集(dataset)的分類器。這次的 CIFAR-10 是一個比起 MNIST 手寫辨識來得更難一些的題目。除了圖片的尺寸變成了 32 x 32 之外,CIFAR-10 也不再是單純的灰階值,而是有著 RGB 三原色的圖片了。

Read More »[PyTorch] Getting Start: 訓練 CIFAR-10 資料集的分類模型

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

Last Updated on 2021-04-13 by Clay

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

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

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

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