New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Expand categorical node. #6028
Expand categorical node. #6028
Conversation
The test is quite limited as there's no way to build categorical splits right now except for manually expanding nodes. So I will leave better tests for later PRs. |
Codecov Report
@@ Coverage Diff @@
## master #6028 +/- ##
=======================================
Coverage 79.10% 79.10%
=======================================
Files 12 12
Lines 3044 3044
=======================================
Hits 2408 2408
Misses 636 636
Continue to review full report at Codecov.
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM. Some minor comments.
Co-authored-by: Philip Hyunsu Cho <chohyu01@cs.washington.edu>
This PR brought significant impact on performance as it adds 1 extra vector for each node. I will work on another PR to restore the performance as I can't revert it cleanly. |
This PR includes expanding tree on categorical feature, saving/loading such trees. It's partly extracted from #5949 , with changes that now the saved tree model contains actual categories instead of bitsets.