Skip to content

20 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 資料集的分類模型
Exit mobile version