1. 程式人生 > >機器學習5:邏輯迴歸之多分類Multi-class classification

機器學習5:邏輯迴歸之多分類Multi-class classification

上一節講解了邏輯迴歸中的二分類問題的原理與步驟,本節講解多分類問題。

以三個class分類為例,過程如圖所示:

原理性推導省略。

1、如圖1所示,對於每個類別,各分配一個線性模型,通過softmax處理得到每個類別的輸出概率y,且所有y的和等於1;

2、如圖2所示,輸出的y與實際樣本\hat y求交叉熵。

3、最後,將每個訓練樣本的交叉熵相加求得最小值,得到訓練引數。