資源簡介
ID3決策樹的Python代碼和相關的數據集
代碼片段和文件信息
#coding:utf-8
import?numpy
import?math
import?matplotlib.pyplot?as?plt
import?pre_processing
‘‘‘
決策樹的節點結構
attrPos;整型,節點分枝屬性的下標,也就是作為父節點產生分枝所用的屬性
attrValue:字符串型,節點分枝屬性的值,也就是作為子節點從父節點分枝所對應的屬性值
classLabel:字符串型,落入該節點的樣本中比例最高的類標簽
sons:list對象,保存當前節點的所有子節點
‘‘‘
class?Node?:
????attrPos?=?None
????attrValue?=?None
????classLabel?=?None
????def?__init__(self):
????????self.sons?=?[]
‘‘‘
函數功能:計算字典對象所有值的和
變量說明:
inputDict:輸入的字典對象
返回值說明:
返回字典對象所有值的和
‘‘‘
def?calcTotalValue(inputDict)?:
????return?numpy.array(inputDict.values()).sum()
‘‘‘
函數功能:計算當前節點的信息熵
變量說明:
classCount:字典對象,保存當前節點中各個類標簽的出現次數
返回值說明:
ent:float型,返回節點的信息熵
‘‘‘
def?calcEnt(classCount)?:
????ent?=?0.0
????
????#空字典,也就是沒有樣本落入當前節點
????if?classCount?==?{}?:
????????return?ent
????
????valueArray?=?
?屬性????????????大小?????日期????時間???名稱
-----------?---------??----------?-----??----
?????文件????????299??2015-01-04?19:38??agaricus-lepiota\agaricus-lepiota-names.txt
?????文件??????12465??2014-12-29?19:19??agaricus-lepiota\agaricus-lepiota-test.data
?????文件?????361238??2014-12-29?19:19??agaricus-lepiota\agaricus-lepiota-train.data
?????文件????????384??2014-12-29?18:47??decision_tree\.project
?????文件????????593??2015-01-05?10:37??decision_tree\.pydevproject
?????文件????????179??2014-12-29?18:52??decision_tree\.settings\org.eclipse.core.resources.prefs
?????文件??????11104??2015-01-05?10:43??decision_tree\src\decision_tree\decision_tree.py
?????文件???????6558??2015-01-05?10:43??decision_tree\src\decision_tree\decision_tree.pyc
?????文件????????790??2015-01-05?10:43??decision_tree\src\decision_tree\main.py
?????文件????????463??2014-12-29?19:43??decision_tree\src\decision_tree\pre_processing.py
?????文件????????672??2014-12-30?10:36??decision_tree\src\decision_tree\pre_processing.pyc
?????目錄??????????0??2015-01-05?11:21??decision_tree\src\decision_tree
?????目錄??????????0??2015-01-05?11:21??decision_tree\.settings
?????目錄??????????0??2015-01-05?11:21??decision_tree\src
?????目錄??????????0??2015-01-05?11:22??agaricus-lepiota
?????目錄??????????0??2015-01-05?11:21??decision_tree
-----------?---------??----------?-----??----
???????????????394745????????????????????16
評論
共有 條評論