Skip to content

Commit

Permalink
Update Honestprune for Oblique Trees
Browse files Browse the repository at this point in the history
  • Loading branch information
@YuxinB
YuxinB committed Feb 7, 2025
1 parent ab12ca9 commit 2d31df5
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion treeple/tree/honesty/_honest_prune.pyx
View file
Original file line number Diff line number Diff line change
Expand Up @@ -153,7 +153,7 @@ cdef class HonestPruner(Splitter):

n_missing += 1
current_end -= 1
elif p > pos and X_ndarray[sample_idx, feature] <= threshold:
elif p > pos and (self.tree._compute_feature(X_ndarray, sample_idx,&self.tree.nodes[node_idx])<= threshold):
self.samples[p], self.samples[pos] = \
self.samples[pos], self.samples[p]
pos += 1
Expand Down

1 comment on commit 2d31df5

@YuxinB
Copy link
Member Author

@YuxinB YuxinB commented on 2d31df5 Feb 7, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Based on main branch, modify pruner for oblique tree

Please sign in to comment.