1. 程式人生 > >機器學習之--使用決策樹分類測試數據(遞歸)

機器學習之--使用決策樹分類測試數據(遞歸)

-- 分類 測試數據 label dict 決策 序號 特征 span

def classfify(mytree,testdata,labels):
    # print(‘mytree:{},testdata:{},labels:{}‘.format(mytree,testdata,labels))
    firststr = list(mytree.keys())[0]                   #樹中的第一個keys值獲取,即第一個特征值名稱
    # print(‘firststr:‘,firststr)
    index = labels.index(firststr)                      #第一個特征值對應labels中的序號
second_dict = mytree[firststr] #第一個特征值對應的value字典 value = testdata[index] third = second_dict[value] #value值後 下一個特征值(字典 or 結果) if isinstance(third,dict): #如果是字典,繼續遞歸 return classfify(third,testdata,labels)
else: #否則則是結果,直接返回 return third

機器學習之--使用決策樹分類測試數據(遞歸)