Skip to content

Commit 65c9b9d

Browse files
author
[zebinyang]
committed
change the way of spreading grid points; version 0.2.0
1 parent 8332491 commit 65c9b9d

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

simtree/mobtree.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -117,8 +117,6 @@ def screen_features(self, sample_indice):
117117
elif n_samples > 2 * self.min_samples_leaf:
118118
if (i + 1 - self.min_samples_leaf) / (n_samples - 2 * self.min_samples_leaf) < split_point / (self.n_screen_grid - 1):
119119
continue
120-
elif (i + 1) != self.min_samples_leaf:
121-
continue
122120

123121
split_point += 1
124122
left_indice = sortted_indice[:(i + 1)]
@@ -184,8 +182,6 @@ def node_split(self, sample_indice):
184182
elif n_samples > 2 * self.min_samples_leaf:
185183
if (i + 1 - self.min_samples_leaf) / (n_samples - 2 * self.min_samples_leaf) < split_point / (self.n_split_grid - 1):
186184
continue
187-
elif (i + 1) != self.min_samples_leaf:
188-
continue
189185

190186
split_point += 1
191187
left_indice = sortted_indice[:(i + 1)]

0 commit comments

Comments
 (0)