Skip to content

Commit 672a82a

Browse files
authored
Update cart.py
1 parent a76fd0f commit 672a82a

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

charpter15_random_forest/cart.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ def __init__(self, min_samples_split=2, min_gini_impurity=999,
4141
def fit(self, X, y, loss=None):
4242
# 递归构建决策树
4343
self.root = self._build_tree(X, y)
44-
self.loss=None
44+
self.loss = None
4545

4646
### 决策树构建函数
4747
def _build_tree(self, X, y, current_depth=0):
@@ -129,8 +129,7 @@ def predict(self, X):
129129
y_pred = [self.predict_value(sample) for sample in X]
130130
return y_pred
131131

132-
133-
132+
134133
class ClassificationTree(BinaryDecisionTree):
135134
### 定义基尼不纯度计算过程
136135
def _calculate_gini_impurity(self, y, y1, y2):

0 commit comments

Comments
 (0)