根據基尼指數生成決策樹代碼

根據基尼指數生成決策樹時,要注意幾個問題:node 1、須要判斷數據是分類型的仍是數值型的,對分類型的拆分子集需判斷值是否相等,而對數值型的則須要先計算出區分的值,比較數據與區分值的大小進行拆分。python 2、若是是分類型還要注意一個問題,由於基尼指數生成的都是二叉樹。因此當類別大於3時,狀況會比較多,好比特徵的取值有4個的話,須要分紅6類。但本例子,是按照特徵取值最多爲3類的狀況來寫的。若是
相關文章
相關標籤/搜索