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

Support multiple treatments in CausalTreeRegressor #775

Open
jeongyoonlee opened this issue May 10, 2024 · 0 comments
Open

Support multiple treatments in CausalTreeRegressor #775

jeongyoonlee opened this issue May 10, 2024 · 0 comments
Labels
enhancement New feature or request

Comments

@jeongyoonlee
Copy link
Collaborator

Is your feature request related to a problem? Please describe.
Currently, CausalTreeRegressor doesn't support multiple treatments, unlike UpliftTreeClassifier or meta-learners.

Describe the solution you'd like
We can follow the implementation in UpliftTreeClassifier, which evaluates splitting criteria across all treatment groups at each node.

Describe alternatives you've considered
Training one CausalTreeRegressor per each treatment separately, then combining the results. This will be a quick and dirty solution.

Additional context

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