Skip to content
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

Define core multi-target regression tree structure. #8884

Merged
merged 5 commits into from
Mar 9, 2023

Conversation

trivialfis
Copy link
Member

@trivialfis trivialfis commented Mar 8, 2023

  • Define a new tree struct embedded in the RegTree.
  • Provide dispatching functions in RegTree.
  • Fix some c++-17 warnings about the use of nodiscard (currently we disable the warning on the CI).
  • Use uint32_t instead of size_t for bst_target_t as it has a defined size and can be used as part of a dmlc parameter.
  • Hide the Segment struct inside the categorical split matrix.
  • No tree statistics yet. Other fields like categorical splits will continue to be part of the original tree structure.

Extracted from #8616 .

- Define a new tree struct embedded in the `RegTree`.
- Provide dispatching functions in `RegTree`.
- Fix some c++-17 warnings about the use of nodiscard (currently we disable the warning on
  the CI).
- Use uint32_t instead of size_t for `bst_target_t` as it has defined size and can be used
  as part of dmlc parameter.
- Hide the `Segment` struct inside categorical split matrix.
@trivialfis trivialfis merged commit 5feee8d into dmlc:master Mar 9, 2023
@trivialfis trivialfis deleted the multi-target-regtree branch March 9, 2023 11:03
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants