search

wepe / MachineLearning

Basic Machine Learning and Deep Learning
5.07k stars 3.16k forks source link

decision Tree 有bug 啊 #25

Open weishengkui opened 6 years ago

weishengkui commented 6 years ago

def _createTree(self,X,y,featureIndex): 有一个返回分支 #所有label都相同的话，则停止分割，返回该label if labelList.count(labelList[0]) == len(labelList): return labelList[0] 当输入数据所有的类别都相同的时候，myTree 被y值覆盖了。