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

response types for leaf nodes #5

Open
3 tasks
GStechschulte opened this issue Aug 30, 2024 · 0 comments
Open
3 tasks

response types for leaf nodes #5

GStechschulte opened this issue Aug 30, 2024 · 0 comments
Labels
enhancement New feature or request

Comments

@GStechschulte
Copy link
Owner

GStechschulte commented Aug 30, 2024

The original PyMC-BART implementation supports the following leaf node response predictions:

  • constant (mean value)
  • linear (linear regression)
  • mixed (not a priority)

These should be implemented as traits and or enums.

We should also make the interface such that response is easily extensible, e.g to add MOTR-BART, TV-BART, etc.

@GStechschulte GStechschulte added the enhancement New feature or request label Aug 30, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

1 participant