Skip to content

Support multiple treatments in CausalTreeRegressor #775

@jeongyoonlee

Description

@jeongyoonlee

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

Metadata

Metadata

Assignees

Labels

enhancementNew feature or request

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions