CS_ALearn/ailearning
1
0
Fork 0
mirror of https://github.com/apachecn/ailearning.git synced 2026-02-09 13:25:39 +08:00
Code Issues Projects Releases Wiki Activity

添加工具类信息增益的计算

Browse Source
This commit is contained in:
jiangzhonglian
2017-03-01 20:33:29 +08:00
parent 5ad0a5099b
commit 0853772faa
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/python/tools/DecisionTree.py → src/python/tools/DecisionTree_getInfoGain.py
View File

@@ -120,5 +120,5 @@ if __name__ == '__main__':
infoGain1 = getFeatureShannonEnt(dataSet1, labels)
infoGain2 = getFeatureShannonEnt(dataSet2, labels)
infoGain3 = getFeatureShannonEnt(dataSet3, labels)
print '香农熵: \n\t%s, \n\t%s, \n\t%s' % (infoGain1, infoGain2, infoGain3)
print '信息增益: \n\t%s, \n\t%s, \n\t%s' % (infoGain1, infoGain2, infoGain3)