Update Decision Tree.py

Jack-Cherish · Jul 18, 2018 · 357c2cd · 357c2cd
1 parent 4a2d47d
commit 357c2cd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Decision Tree/Decision Tree.py b/Decision Tree/Decision Tree.py
@@ -25,7 +25,7 @@ def calcShannonEnt(dataSet):
 	for featVec in dataSet:							#对每组特征向量进行统计
 		currentLabel = featVec[-1]					#提取标签(Label)信息
 		if currentLabel not in labelCounts.keys():	#如果标签(Label)没有放入统计次数的字典,添加进去
-			labelCounts[currentLabel] = 1
+			labelCounts[currentLabel] = 0
 		labelCounts[currentLabel] += 1				#Label计数
 	shannonEnt = 0.0								#经验熵(香农熵)
 	for key in labelCounts:							#计算香农熵