決策樹(shù)(Decision Tree)在機(jī)器學(xué)習(xí)中也是比較常見(jiàn)的一種算法,最早的決策樹(shù)算法是ID3,改善后得到了C4.5算法,進(jìn)一步改進(jìn)后形成了我們現(xiàn)在使用的C5.0算法,綜合性能大幅提高。 算法核心:為每一次分裂確定一個(gè)分裂屬性。ID3采用的是“信息增益”為度量來(lái)選擇分裂屬性的。 本文在Excel中建模進(jìn)行決策樹(shù)分析,屬于基礎(chǔ)的決策樹(shù)學(xué)習(xí),有興趣的可以在SPSS Modeler和Python中進(jìn)行操作。 樹(shù)模型(又稱(chēng)決策樹(shù)或者樹(shù)結(jié)構(gòu)模型):基本思想和方差分析中的變異分解極為相似。 決策樹(shù)圖例 現(xiàn)在我們來(lái)分析天氣、溫度、濕度、風(fēng)這些屬性對(duì)打球的影響 首先確定樣本集信息熵,然后計(jì)算各個(gè)屬性的信息增益進(jìn)行對(duì)比分析。 熵:數(shù)據(jù)集中的不確定性、突發(fā)性或隨機(jī)性的程度的度量。當(dāng)一個(gè)數(shù)據(jù)集中的記錄全部都屬于同一類(lèi)的時(shí)候,則沒(méi)有不確定性,此時(shí)熵為0。 信息增益:按照某個(gè)屬性A把數(shù)據(jù)集S分裂,所得到的信息增益等于數(shù)據(jù)集S的熵減去各個(gè)子集的熵的加權(quán)和。 計(jì)算是否打球的概率: 計(jì)算天氣對(duì)打球的影響: I(晴天)=-0.25*log(0.25,2)-0.75*log(0.75,2)=0.811278 E(天氣)=0.285714*0.811278+0.357143*0+0.357143*0.70951=0.578562 Gain(天氣)=E(all)-E(天氣)=0.940286-0.578562=0.361724 經(jīng)計(jì)算得出天氣的信息增益為0.361724,溫度、濕度和風(fēng)計(jì)算步驟類(lèi)似
對(duì)各屬性的信息增益進(jìn)行降序排序,選擇最大的作為分裂屬性 |
|