python實現(xiàn)決策樹算法-創(chuàng)新互聯(lián)

python 實現(xiàn)決策樹算法？相信很多沒有經(jīng)驗的人對此束手無策，為此本文總結(jié)了問題出現(xiàn)的原因和解決方法，通過這篇文章希望你能解決這個問題。

創(chuàng)新互聯(lián)建站自2013年起，先為鼓樓等服務(wù)建站，鼓樓等地企業(yè)，進(jìn)行企業(yè)商務(wù)咨詢服務(wù)。為鼓樓企業(yè)網(wǎng)站制作PC+手機(jī)+微官網(wǎng)三網(wǎng)同步一站式服務(wù)解決您的所有建站問題。
'''
數(shù)據(jù)集：Mnist
訓(xùn)練集數(shù)量：60000
測試集數(shù)量：10000
------------------------------
運(yùn)行結(jié)果：ID3(未剪枝)
  正確率：85.9%
  運(yùn)行時長：356s
'''

import time
import numpy as np


def loadData(fileName):
  '''
  加載文件
  :param fileName:要加載的文件路徑
  :return: 數(shù)據(jù)集和標(biāo)簽集
  '''
  # 存放數(shù)據(jù)及標(biāo)記
  dataArr = [];
  labelArr = []
  # 讀取文件
  fr = open(fileName)
  # 遍歷文件中的每一行
  for line in fr.readlines():
    # 獲取當(dāng)前行，并按“，”切割成字段放入列表中
    # strip：去掉每行字符串首尾指定的字符（默認(rèn)空格或換行符）
    # split：按照指定的字符將字符串切割成每個字段，返回列表形式
    curLine = line.strip().split(',')
    # 將每行中除標(biāo)記外的數(shù)據(jù)放入數(shù)據(jù)集中（curLine[0]為標(biāo)記信息）
    # 在放入的同時將原先字符串形式的數(shù)據(jù)轉(zhuǎn)換為整型
    # 此外將數(shù)據(jù)進(jìn)行了二值化處理，大于128的轉(zhuǎn)換成1，小于的轉(zhuǎn)換成0，方便后續(xù)計算
    dataArr.append([int(int(num) > 128) for num in curLine[1:]])
    # 將標(biāo)記信息放入標(biāo)記集中
    # 放入的同時將標(biāo)記轉(zhuǎn)換為整型
    labelArr.append(int(curLine[0]))
  # 返回數(shù)據(jù)集和標(biāo)記
  return dataArr, labelArr


def majorClass(labelArr):
  '''
  找到當(dāng)前標(biāo)簽集中占數(shù)目大的標(biāo)簽
  :param labelArr: 標(biāo)簽集
  :return: 大的標(biāo)簽
  '''
  # 建立字典，用于不同類別的標(biāo)簽技術(shù)
  classDict = {}
  # 遍歷所有標(biāo)簽
  for i in range(len(labelArr)):
    # 當(dāng)?shù)谝淮斡龅紸標(biāo)簽時，字典內(nèi)還沒有A標(biāo)簽，這時候直接幅值加1是錯誤的，
    # 所以需要判斷字典中是否有該鍵，沒有則創(chuàng)建，有就直接自增
    if labelArr[i] in classDict.keys():
      # 若在字典中存在該標(biāo)簽，則直接加1
      classDict[labelArr[i]] += 1
    else:
      # 若無該標(biāo)簽，設(shè)初值為1，表示出現(xiàn)了1次了
      classDict[labelArr[i]] = 1
  # 對字典依據(jù)值進(jìn)行降序排序
  classSort = sorted(classDict.items(), key=lambda x: x[1], reverse=True)
  # 返回大一項的標(biāo)簽，即占數(shù)目最多的標(biāo)簽
  return classSort[0][0]


def calc_H_D(trainLabelArr):
  '''
  計算數(shù)據(jù)集D的經(jīng)驗熵，參考公式5.7 經(jīng)驗熵的計算
  :param trainLabelArr:當(dāng)前數(shù)據(jù)集的標(biāo)簽集
  :return: 經(jīng)驗熵
  '''
  # 初始化為0
  H_D = 0
  # 將當(dāng)前所有標(biāo)簽放入集合中，這樣只要有的標(biāo)簽都會在集合中出現(xiàn)，且出現(xiàn)一次。
  # 遍歷該集合就可以遍歷所有出現(xiàn)過的標(biāo)記并計算其Ck
  # 這么做有一個很重要的原因：首先假設(shè)一個背景，當(dāng)前標(biāo)簽集中有一些標(biāo)記已經(jīng)沒有了，比如說標(biāo)簽集中
  # 沒有0（這是很正常的，說明當(dāng)前分支不存在這個標(biāo)簽）。 式5.7中有一項Ck，那按照式中的針對不同標(biāo)簽k
  # 計算Cl和D并求和時，由于沒有0，那么C0=0，此時C0/D0=0,log2(C0/D0) = log2(0)，事實上0并不在log的
  # 定義區(qū)間內(nèi)，出現(xiàn)了問題
  # 所以使用集合的方式先知道當(dāng)前標(biāo)簽中都出現(xiàn)了那些標(biāo)簽，隨后對每個標(biāo)簽進(jìn)行計算，如果沒出現(xiàn)的標(biāo)簽?zāi)且豁椌?  # 不在經(jīng)驗熵中出現(xiàn)（未參與，對經(jīng)驗熵?zé)o影響），保證log的計算能一直有定義
  trainLabelSet = set([label for label in trainLabelArr])
  # 遍歷每一個出現(xiàn)過的標(biāo)簽
  for i in trainLabelSet:
    # 計算|Ck|/|D|
    # trainLabelArr == i：當(dāng)前標(biāo)簽集中為該標(biāo)簽的的位置
    # 例如a = [1, 0, 0, 1], c = (a == 1): c == [True, false, false, True]
    # trainLabelArr[trainLabelArr == i]：獲得為指定標(biāo)簽的樣本
    # trainLabelArr[trainLabelArr == i].size：獲得為指定標(biāo)簽的樣本的大小，即標(biāo)簽為i的樣本
    # 數(shù)量，就是|Ck|
    # trainLabelArr.size：整個標(biāo)簽集的數(shù)量（也就是樣本集的數(shù)量），即|D|
    p = trainLabelArr[trainLabelArr == i].size / trainLabelArr.size
    # 對經(jīng)驗熵的每一項累加求和
    H_D += -1 * p * np.log2(p)

  # 返回經(jīng)驗熵
  return H_D


def calcH_D_A(trainDataArr_DevFeature, trainLabelArr):
  '''
  計算經(jīng)驗條件熵
  :param trainDataArr_DevFeature:切割后只有feature那列數(shù)據(jù)的數(shù)組
  :param trainLabelArr: 標(biāo)簽集數(shù)組
  :return: 經(jīng)驗條件熵
  '''
  # 初始為0
  H_D_A = 0
  # 在featue那列放入集合中，是為了根據(jù)集合中的數(shù)目知道該feature目前可取值數(shù)目是多少
  trainDataSet = set([label for label in trainDataArr_DevFeature])

  # 對于每一個特征取值遍歷計算條件經(jīng)驗熵的每一項
  for i in trainDataSet:
    # 計算H(D|A)
    # trainDataArr_DevFeature[trainDataArr_DevFeature == i].size / trainDataArr_DevFeature.size:|Di| / |D|
    # calc_H_D(trainLabelArr[trainDataArr_DevFeature == i]):H(Di)
    H_D_A += trainDataArr_DevFeature[trainDataArr_DevFeature == i].size / trainDataArr_DevFeature.size \
         * calc_H_D(trainLabelArr[trainDataArr_DevFeature == i])
  # 返回得出的條件經(jīng)驗熵
  return H_D_A


def calcBestFeature(trainDataList, trainLabelList):
  '''
  計算信息增益大的特征
  :param trainDataList: 當(dāng)前數(shù)據(jù)集
  :param trainLabelList: 當(dāng)前標(biāo)簽集
  :return: 信息增益大的特征及大信息增益值
  '''
  # 將數(shù)據(jù)集和標(biāo)簽集轉(zhuǎn)換為數(shù)組形式
  # trainLabelArr轉(zhuǎn)換后需要轉(zhuǎn)置，這樣在取數(shù)時方便
  # 例如a = np.array([1, 2, 3]); b = np.array([1, 2, 3]).T
  # 若不轉(zhuǎn)置，a[0] = [1, 2, 3]，轉(zhuǎn)置后b[0] = 1, b[1] = 2
  # 對于標(biāo)簽集來說，能夠很方便地取到每一位是很重要的
  trainDataArr = np.array(trainDataList)
  trainLabelArr = np.array(trainLabelList).T

  # 獲取當(dāng)前特征數(shù)目，也就是數(shù)據(jù)集的橫軸大小
  featureNum = trainDataArr.shape[1]

  # 初始化大信息增益
  maxG_D_A = -1
  # 初始化大信息增益的特征
  maxFeature = -1
  # 對每一個特征進(jìn)行遍歷計算
  for feature in range(featureNum):
    # “5.2.2 信息增益”中“算法5.1（信息增益的算法）”第一步：
    # 1.計算數(shù)據(jù)集D的經(jīng)驗熵H(D)
    H_D = calc_H_D(trainLabelArr)
    # 2.計算條件經(jīng)驗熵H(D|A)
    # 由于條件經(jīng)驗熵的計算過程中只涉及到標(biāo)簽以及當(dāng)前特征，為了提高運(yùn)算速度（全部樣本
    # 做成的矩陣運(yùn)算速度太慢，需要剔除不需要的部分），將數(shù)據(jù)集矩陣進(jìn)行切割
    # 數(shù)據(jù)集在初始時刻是一個Arr = 60000*784的矩陣，針對當(dāng)前要計算的feature，在訓(xùn)練集中切割下
    # Arr[:, feature]這么一條來，因為后續(xù)計算中數(shù)據(jù)集中只用到這個（沒明白的跟著算一遍例5.2）
    # trainDataArr[:, feature]:在數(shù)據(jù)集中切割下這么一條
    # trainDataArr[:, feature].flat：將這么一條轉(zhuǎn)換成豎著的列表
    # np.array(trainDataArr[:, feature].flat)：再轉(zhuǎn)換成一條豎著的矩陣，大小為60000*1（只是初始是
    # 這么大，運(yùn)行過程中是依據(jù)當(dāng)前數(shù)據(jù)集大小動態(tài)變的）
    trainDataArr_DevideByFeature = np.array(trainDataArr[:, feature].flat)
    # 3.計算信息增益G(D|A)  G(D|A) = H(D) - H(D | A)
    G_D_A = H_D - calcH_D_A(trainDataArr_DevideByFeature, trainLabelArr)
    # 不斷更新大的信息增益以及對應(yīng)的feature
    if G_D_A > maxG_D_A:
      maxG_D_A = G_D_A
      maxFeature = feature
  return maxFeature, maxG_D_A


def getSubDataArr(trainDataArr, trainLabelArr, A, a):
  '''
  更新數(shù)據(jù)集和標(biāo)簽集
  :param trainDataArr:要更新的數(shù)據(jù)集
  :param trainLabelArr: 要更新的標(biāo)簽集
  :param A: 要去除的特征索引
  :param a: 當(dāng)data[A]== a時，說明該行樣本時要保留的
  :return: 新的數(shù)據(jù)集和標(biāo)簽集
  '''
  # 返回的數(shù)據(jù)集
  retDataArr = []
  # 返回的標(biāo)簽集
  retLabelArr = []
  # 對當(dāng)前數(shù)據(jù)的每一個樣本進(jìn)行遍歷
  for i in range(len(trainDataArr)):
    # 如果當(dāng)前樣本的特征為指定特征值a
    if trainDataArr[i][A] == a:
      # 那么將該樣本的第A個特征切割掉，放入返回的數(shù)據(jù)集中
      retDataArr.append(trainDataArr[i][0:A] + trainDataArr[i][A + 1:])
      # 將該樣本的標(biāo)簽放入返回標(biāo)簽集中
      retLabelArr.append(trainLabelArr[i])
  # 返回新的數(shù)據(jù)集和標(biāo)簽集
  return retDataArr, retLabelArr


def createTree(*dataSet):
  '''
  遞歸創(chuàng)建決策樹
  :param dataSet:(trainDataList， trainLabelList) <<-- 元祖形式
  :return:新的子節(jié)點(diǎn)或該葉子節(jié)點(diǎn)的值
  '''
  # 設(shè)置Epsilon，“5.3.1 ID3算法”第4步提到需要將信息增益與閾值Epsilon比較，若小于則直接處理后返回T
  Epsilon = 0.1
  # 從參數(shù)中獲取trainDataList和trainLabelList
  trainDataList = dataSet[0][0]
  trainLabelList = dataSet[0][1]
  # 打印信息：開始一個子節(jié)點(diǎn)創(chuàng)建，打印當(dāng)前特征向量數(shù)目及當(dāng)前剩余樣本數(shù)目
  print('start a node', len(trainDataList[0]), len(trainLabelList))

  # 將標(biāo)簽放入一個字典中，當(dāng)前樣本有多少類，在字典中就會有多少項
  # 也相當(dāng)于去重，多次出現(xiàn)的標(biāo)簽就留一次。舉個例子，假如處理結(jié)束后字典的長度為1，那說明所有的樣本
  # 都是同一個標(biāo)簽，那就可以直接返回該標(biāo)簽了，不需要再生成子節(jié)點(diǎn)了。
  classDict = {i for i in trainLabelList}
  # 如果D中所有實例屬于同一類Ck，則置T為單節(jié)點(diǎn)數(shù)，并將Ck作為該節(jié)點(diǎn)的類，返回T
  # 即若所有樣本的標(biāo)簽一致，也就不需要再分化，返回標(biāo)記作為該節(jié)點(diǎn)的值，返回后這就是一個葉子節(jié)點(diǎn)
  if len(classDict) == 1:
    # 因為所有樣本都是一致的，在標(biāo)簽集中隨便拿一個標(biāo)簽返回都行，這里用的第0個（因為你并不知道
    # 當(dāng)前標(biāo)簽集的長度是多少，但運(yùn)行中所有標(biāo)簽只要有長度都會有第0位。
    return trainLabelList[0]

  # 如果A為空集，則置T為單節(jié)點(diǎn)數(shù)，并將D中實例數(shù)大的類Ck作為該節(jié)點(diǎn)的類，返回T
  # 即如果已經(jīng)沒有特征可以用來再分化了，就返回占大多數(shù)的類別
  if len(trainDataList[0]) == 0:
    # 返回當(dāng)前標(biāo)簽集中占數(shù)目大的標(biāo)簽
    return majorClass(trainLabelList)

  # 否則，按式5.10計算A中個特征值的信息增益，選擇信息增益大的特征Ag
  Ag, EpsilonGet = calcBestFeature(trainDataList, trainLabelList)

  # 如果Ag的信息增益比小于閾值Epsilon，則置T為單節(jié)點(diǎn)樹，并將D中實例數(shù)大的類Ck
  # 作為該節(jié)點(diǎn)的類，返回T
  if EpsilonGet < Epsilon:
    return majorClass(trainLabelList)

  # 否則，對Ag的每一可能值ai，依Ag=ai將D分割為若干非空子集Di，將Di中實例數(shù)大的
  # 類作為標(biāo)記，構(gòu)建子節(jié)點(diǎn)，由節(jié)點(diǎn)及其子節(jié)點(diǎn)構(gòu)成樹T，返回T
  treeDict = {Ag: {}}
  # 特征值為0時，進(jìn)入0分支
  # getSubDataArr(trainDataList, trainLabelList, Ag, 0)：在當(dāng)前數(shù)據(jù)集中切割當(dāng)前feature，返回新的數(shù)據(jù)集和標(biāo)簽集
  treeDict[Ag][0] = createTree(getSubDataArr(trainDataList, trainLabelList, Ag, 0))
  treeDict[Ag][1] = createTree(getSubDataArr(trainDataList, trainLabelList, Ag, 1))

  return treeDict


def predict(testDataList, tree):
  '''
  預(yù)測標(biāo)簽
  :param testDataList:樣本
  :param tree: 決策樹
  :return: 預(yù)測結(jié)果
  '''
  # treeDict = copy.deepcopy(tree)

  # 死循環(huán)，直到找到一個有效地分類
  while True:
    # 因為有時候當(dāng)前字典只有一個節(jié)點(diǎn)
    # 例如{73: {0: {74:6}}}看起來節(jié)點(diǎn)很多，但是對于字典的最頂層來說，只有73一個key，其余都是value
    # 若還是采用for來讀取的話不太合適，所以使用下行這種方式讀取key和value
    (key, value), = tree.items()
    # 如果當(dāng)前的value是字典，說明還需要遍歷下去
    if type(tree[key]).__name__ == 'dict':
      # 獲取目前所在節(jié)點(diǎn)的feature值，需要在樣本中刪除該feature
      # 因為在創(chuàng)建樹的過程中，feature的索引值永遠(yuǎn)是對于當(dāng)時剩余的feature來設(shè)置的
      # 所以需要不斷地刪除已經(jīng)用掉的特征，保證索引相對位置的一致性
      dataVal = testDataList[key]
      del testDataList[key]
      # 將tree更新為其子節(jié)點(diǎn)的字典
      tree = value[dataVal]
      # 如果當(dāng)前節(jié)點(diǎn)的子節(jié)點(diǎn)的值是int，就直接返回該int值
      # 例如{403: {0: 7, 1: {297:7}}，dataVal=0
      # 此時上一行tree = value[dataVal]，將tree定位到了7，而7不再是一個字典了，
      # 這里就可以直接返回7了，如果tree = value[1]，那就是一個新的子節(jié)點(diǎn)，需要繼續(xù)遍歷下去
      if type(tree).__name__ == 'int':
        # 返回該節(jié)點(diǎn)值，也就是分類值
        return tree
    else:
      # 如果當(dāng)前value不是字典，那就返回分類值
      return value


def accuracy(testDataList, testLabelList, tree):
  '''
  測試準(zhǔn)確率
  :param testDataList:待測試數(shù)據(jù)集
  :param testLabelList: 待測試標(biāo)簽集
  :param tree: 訓(xùn)練集生成的樹
  :return: 準(zhǔn)確率
  '''
  # 錯誤次數(shù)計數(shù)
  errorCnt = 0
  # 遍歷測試集中每一個測試樣本
  for i in range(len(testDataList)):
    # 判斷預(yù)測與標(biāo)簽中結(jié)果是否一致
    if testLabelList[i] != predict(testDataList[i], tree):
      errorCnt += 1
  # 返回準(zhǔn)確率
  return 1 - errorCnt / len(testDataList)


if __name__ == '__main__':
  # 開始時間
  start = time.time()

  # 獲取訓(xùn)練集
  trainDataList, trainLabelList = loadData('../Mnist/mnist_train.csv')
  # 獲取測試集
  testDataList, testLabelList = loadData('../Mnist/mnist_test.csv')

  # 創(chuàng)建決策樹
  print('start create tree')
  tree = createTree((trainDataList, trainLabelList))
  print('tree is:', tree)

  # 測試準(zhǔn)確率
  print('start test')
  accur = accuracy(testDataList, testLabelList, tree)
  print('the accur is:', accur)

  # 結(jié)束時間
  end = time.time()
  print('time span:', end - start)

新聞名稱：python實現(xiàn)決策樹算法-創(chuàng)新互聯(lián)
本文來源：http://bm7419.com/article30/cdgdpo.html

成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián)，為您提供搜索引擎優(yōu)化、網(wǎng)站收錄、ChatGPT、微信公眾號、品牌網(wǎng)站建設(shè)、做網(wǎng)站

廣告

聲明：本網(wǎng)站發(fā)布的內(nèi)容（圖片、視頻和文字）以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主，如果涉及侵權(quán)請盡快告知，我們將會在第一時間刪除。文章觀點(diǎn)不代表本網(wǎng)站立場，如需處理請聯(lián)系客服。電話：028-86922220；郵箱：631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載，或轉(zhuǎn)載時需注明來源： 創(chuàng)新互聯(lián)

成都定制網(wǎng)站建設(shè)