Skip to content

Commit 254f334

Browse files
author
[zebinyang]
committed
fix a bug in CART Regressor and Classifier (sum_left update should be executed before checking min_samples_leaf); version 0.2.3
1 parent 97e8347 commit 254f334

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

simtree/mobtree.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -307,6 +307,7 @@ def fit(self, x, y):
307307
"depth": depth + 1,
308308
"impurity": split["right_impurity"],
309309
"is_left": False})
310+
self.leaf_idx_list = [key for key, item in self.tree.items() if item["is_leaf"]]
310311
return self
311312

312313
def plot_tree(self, draw_depth=np.inf, start_node_id=1, folder="./results/", name="demo", save_png=False, save_eps=False):

0 commit comments

Comments
 (0)