Skip to content

Commit 2d31df5

Browse files
committed
Update Honestprune for Oblique Trees
1 parent ab12ca9 commit 2d31df5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

treeple/tree/honesty/_honest_prune.pyx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -153,7 +153,7 @@ cdef class HonestPruner(Splitter):
153153

154154
n_missing += 1
155155
current_end -= 1
156-
elif p > pos and X_ndarray[sample_idx, feature] <= threshold:
156+
elif p > pos and (self.tree._compute_feature(X_ndarray, sample_idx,&self.tree.nodes[node_idx])<= threshold):
157157
self.samples[p], self.samples[pos] = \
158158
self.samples[pos], self.samples[p]
159159
pos += 1

0 commit comments

Comments
 (0)