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.

Sign up for GitHub

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

Jump to bottom

Support for unordered categorical features #8

Open
3 of 4 tasks
andrewherren opened this issue Oct 31, 2023 · 0 comments
Open
3 of 4 tasks

Support for unordered categorical features #8

andrewherren opened this issue Oct 31, 2023 · 0 comments
Labels
enhancement New feature or request

Comments

@andrewherren
Copy link
Collaborator

andrewherren commented Oct 31, 2023 •
edited
Loading

Supporting unordered categorical features requires the following modifications:

  • TrainData class must keep track of which features are categorical (ideally encoding them with integers for space efficiency)
  • Tree class must be able to represent set-defined categorical splits
  • XBART split sampler cannot generally enumerate all possible categorical splits, so should use the Fisher (1958) approach
  • BART split sampler should use the Deshpande (2023) approach
@andrewherren andrewherren added the enhancement New feature or request label Oct 31, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
enhancement New feature or request
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@andrewherren