-
Notifications
You must be signed in to change notification settings - Fork 831
Open
Labels
enhancementNew feature or requestNew feature or request
Description
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 requestNew feature or request