microsoft / microsoft/SynapseML

Support multiple treatments in Causal ML

Open
#2,328 2 comments 0 reactions 0 assignees View on GitHub
triage
Dominant language
Scala
Stars
5.2k
Forks
868
Avg merge
22h 9m
Merged PRs (30d)
45

Description

**Is your feature request related to a problem? Please describe.**
My causal ML problem has multiple treatment variables. I am trying it with EconML and it works but it needs too much memory - it doesn't scale up to big data. But it looks like SynapseML causal ml only supports a single treatment. Or am I misunderstanding it.

**Describe the solution you'd like**
Many DoubleML techniques support multiple treatments. For example, this is supported in EconML DML classes. Are there any plans to add support for multiple treatments in the SynapseML Causal ML library?

**Additional context**
This would help make the causal ML tools practical in the real world - most non-academic problems have multiple treatment variables.

Contributor guide

Open the contributing guide

Research direction

Start by reading the SynapseML Causal ML library and checking how treatment variables are currently represented and limited to one treatment. Compare that behavior with the multiple-treatment support described for EconML and DoubleML; done would require an agreed design and documented support for multiple treatments at larger data sizes.

Written by the indexing model from the issue text.

Assessment

Tech stack
scala, spark
Domain
data, machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.