python实现决策树

一、决策树原理

决策树是用样本的属性作为结点，用属性的取值作为分支的树结构。 
决策树的根结点是所有样本中信息量最大的属性。树的中间结点是该结点为根的子树所包含的样本子集中信息量最大的属性。决策树的叶结点是样本的类别值。决策树是一种知识表示形式，它是对所有样本数据的高度概括决策树能准确地识别所有样本的类别，也能有效地识别新样本的类别。

 

决策树算法ID3的基本思想：

        首先找出最有判别力的属性，把样例分成多个子集，每个子集又选择最有判别力的属性进行划分，一直进行到所有子集仅包含同一类型的数据为止。最后得到一棵决策树。

       J.R.Quinlan的工作主要是引进了信息论中的信息增益，他将其称为信息增益（information gain），作为属性判别能力的度量，设计了构造决策树的递归算法。

决策树的一般流程：

（1）收集数据：可以使用任何方法。

（2）准备数据：树构造算法只适用于标称型数据，因此数值型数据必须离散化。

（3）分析数据：可以使用任何方法，构造数完成之后，我们应该检查图形是否符合预期。

（4）训练算法：构造树的数据结构。

（5）测试算法：使用经验数计算错误率。

（6）使用算法：此步骤可以适用于任何监督学习算法，而使用决策树可以更好地理解数据的内在含义。

举例子比较容易理解：

对于气候分类问题，属性为:

天气(A₁) 取值为： 晴，多云，雨

气温(A₂)  取值为： 冷 ，适中，热

湿度(A₃)  取值为： 高 ，正常

风 (A₄)  取值为： 有风， 无风

每个样例属于不同的类别，此例仅有两个类别，分别为P，N。P类和N类的样例分别称为正例和反例。将一些已知的正例和反例放在一起便得到训练集。

由ID3算法得出一棵正确分类训练集中每个样例的决策树，见下图。

决策树叶子为类别名，即P 或者N。其它结点由样例的属性组成，每个属性的不同取值对应一分枝。

若要对一样例分类，从树根开始进行测试，按属性的取值分枝向下进入下层结点，对该结点进行测试，过程一直进行到叶结点，样例被判为属于该叶结点所标记的类别。

现用图来判一个具体例子，

某天早晨气候描述为:

天气：多云

气温：冷

湿度：正常

风： 无风

它属于哪类气候呢?-------------从图中可判别该样例的类别为P类。

ID3就是要从表的训练集构造图这样的决策树。实际上，能正确分类训练集的决策树不止一棵。Quinlan的ID3算法能得出结点最少的决策树。

ID3算法：

     ⒈ 对当前例子集合，计算各属性的信息增益；

     ⒉ 选择信息增益最大的属性A_k；

     ⒊ 把在A_k处取值相同的例子归于同一子集，A_k取几个值就得几个子集；

     ⒋ 对既含正例又含反例的子集，递归调用建树算法；

     ⒌ 若子集仅含正例或反例，对应分枝标上P或N，返回调用处。

一般只要涉及到树的情况，经常会要用到递归。

 

对于气候分类问题进行具体计算有：

⒈ 信息熵的计算: 其中S是样例的集合， P(ui)是类别i出现概率：

|S|表示例子集S的总数，|ui|表示类别ui的例子数。对9个正例和5个反例有：

P（u1）=9/14

P（u2）=5/14

H（S）=（9/14）log（14/9）+（5/14）log（14/5）=0.94bit

 

⒉ 信息增益的计算:

其中A是属性，Value(A)是属性A取值的集合，v是A的某一属性值，Sv是S中A的值为v的样例集合，| Sv |为Sv中所含样例数。

以属性A1为例，根据信息增益的计算公式，属性A1的信息增益为

S=[9+,5-] //原样例集中共有14个样例，9个正例，5个反例

S晴=[2+,3-]//属性A1取值晴的样例共5个，2正，3反

S多云=[4+,0-] //属性A1取值多云的样例共4个，4正，0反

S雨=[3+,2-] //属性A1取值晴的样例共5个，3正，2反

故

 

 

3.结果为

属性A1的信息增益最大，所以被选为根结点。

4.建决策树的根和叶子

ID3算法将选择信息增益最大的属性天气作为树根，在14个例子中对天气的3个取值进行分枝，3 个分枝对应3 个子集，分别是:

其中S2中的例子全属于P类，因此对应分枝标记为P，其余两个子集既含有正例又含有反例，将递归调用建树算法。

5.递归建树

分别对S1和S3子集递归调用ID3算法，在每个子集中对各属性求信息增益.

（1）对S1，湿度属性信息增益最大，以它为该分枝的根结点，再向下分枝。湿度取高的例子全为N类，该分枝标记N。取值正常的例子全为P类，该分枝标记P。

（2）对S3，风属性信息增益最大，则以它为该分枝根结点。再向下分枝，风取有风时全为N类，该分枝标记N。取无风时全为P类，该分枝标记P。

二、PYTHON实现决策树案例

                           表一、海洋生物数据集

 

No surfacing(不浮出水面是否可以生存)

Flippers(是否有脚蹼)Fish11（是）1（是）yes21（是）1（是）yes31（是）0（否）no40（否）1（是）no50（否）1（是）no

利用python对表一中的海洋生物数据利用决策树进行分类：

from math import logimport operatordef calcShannonEnt(dataSet):    numEntries = len(dataSet)    labelCounts = {}    for featVec in dataSet:        currentLabel = featVec[-1]        if currentLabel not in labelCounts.keys():            labelCounts[currentLabel] = 0        labelCounts[currentLabel] += 1    shannonEnt = 0.0    for key in labelCounts:        prob = float(labelCounts[key]) / numEntries        shannonEnt -= prob * log(prob, 2)    return shannonEntdef createDataSet():    dataSet = [[1, 1, 'yes'],               [1, 1, 'yes'],               [1, 0, 'no'],               [0, 1, 'no'],               [0, 1, 'no']]    labels = ['no surfacing', 'flippers']    return dataSet, labelsdef splitDataSet(dataSet, axis, value):    retDataSet = []    for featVec in dataSet:        if featVec[axis] == value:            reducedFeatVec = featVec[:axis]            reducedFeatVec.extend(featVec[axis + 1:])            retDataSet.append(reducedFeatVec)    return retDataSetdef chooseBestFeatureToSplit(dataSet):    numFeatures = len(dataSet[0]) - 1    baseEntropy = calcShannonEnt(dataSet)    bestInfoGain = 0.0    bestFeature = -1    for i in range(numFeatures):        featList = [example[i] for example in dataSet]        uniqueVals = set(featList)        newEntropy = 0.0        for value in uniqueVals:            subDataSet = splitDataSet(dataSet, i, value)            prob = len(subDataSet) / float(len(dataSet))            newEntropy += prob * calcShannonEnt(subDataSet)        infoGain = baseEntropy - newEntropy        if (infoGain > bestInfoGain):            bestInfoGain = infoGain            bestFeature = i    return bestFeaturedef majorityCnt(classList):    classCount = {}    for vote in classList:        if vote not in classCount.keys():            classCount[vote] = 0        classCount[vote] += 1    sortedClassCount = sorted(classCount.iteritems(),                              key=operator.itemgetter(1), reverse=True)    return sortedClassCount[0][0]def createTree(dataSet,labels):    classList = [example[-1] for example in dataSet]    if classList.count(classList[0]) == len(classList):        return classList[0]    if len(dataSet[0]) == 1:        return majorityCnt(classList)    bestFeat = chooseBestFeatureToSplit(dataSet)    bestFeatLabel = labels[bestFeat]    myTree = {bestFeatLabel:{}}    del(labels[bestFeat])    featValues = [example[bestFeat] for example in dataSet]    uniqueVals = set(featValues)    for value in uniqueVals:        subLabels = labels[:]        myTree[bestFeatLabel][value] = createTree(splitDataSet(dataSet,bestFeat,value),subLabels)    return myTreemyDat, labels = createDataSet()calcShannonEnt(myDat)splitDataSet(myDat, 0, 1)chooseBestFeatureToSplit(myDat)myTree = createTree(myDat,labels)print myTree

得到的结果如下：

D:\Python27\python.exe D:/Users/Administrator/PycharmProjects/web/DicisionTree/tree.py{'no surfacing': {0: 'no', 1: {'flippers': {0: 'no', 1: 'yes'}}}}