modify

tree/my_tree.py

@@ -127,7 +127,7 @@ MAX_T = 5
 def is_end(Ytrain):
     if len(Ytrain) == 0:
         return True
-    if len(set(Ytrain)) == 1:
+    if len(set(Ytrain)) == 1: # 只有一个分类
         return True
 
 # 强行划分为叶子节点
@@ -166,7 +166,7 @@ def fit(Xtrain, Ytrain, parent_node, depth):
         print('这个是叶子节点')
         return TreeNode(-1, 0, True, -1, len(Ytrain), distrib(Ytrain))
 
-    if depth > MAX_T:
+    if depth >= MAX_T:
         return leaf_node(Ytrain)
 
     i, mean = cal_ent_attr(Xtrain, Ytrain)