Package: causalml
DisclaimerThis project is stable and being incubated for long-term support. It may contain new experimental code, for which APIs are subject to change.Causal ML: A Python Package for Uplift Modeling and Causal Inference with MLCausal MLis a Python package that provides a suite of uplift modeling and causal inference methods using machine learning algorithms based on recent
research[1]. It provides a standard interface that allows user to estimate the Conditional Average Treatment Effect (CATE) or Individual Treatment
Effect (ITE) from experimental or observational data. Essentially, it estimates the causal impact of interventionTon outcomeYfor users
with observed featuresX, without strong assumptions on the model form. Typical use cases includeCampaign targeting optimization: An important lever to increase ROI in an advertising campaign is to target the ad to the set of customers who will have a favorable response in a given KPI such as engagement or sales. CATE identifies these customers by estimating the effect of the KPI from ad exposure at the individual level from A/B experiment or historical observational data.Personalized engagement: A company has multiple options to interact with its customers such as different product choices in up-sell or messaging channels for communications. One can use CATE to estimate the heterogeneous treatment effect for each customer and treatment option combination for an optimal personalized recommendation system.DocumentationDocumentation is available at:https://causalml.readthedocs.io/en/latest/about.htmlInstallationInstallation instructions are available at:https://causalml.readthedocs.io/en/latest/installation.htmlQuickstartQuickstarts with code-snippets are available at:https://causalml.readthedocs.io/en/latest/quickstart.htmlExample NotebooksExample notebooks are available at:https://causalml.readthedocs.io/en/latest/examples.htmlContributingWe welcome community contributors to the project. Before you start, please read ourcode of conductand check outcontributing guidelinesfirst.VersioningWe document versions and changes in ourchangelog.LicenseThis project is licensed under the Apache 2.0 License - see theLICENSEfile for details.ReferencesDocumentationCausal ML API documentationConference Talks and Publications by CausalML Team(Talk) Introduction to CausalML atCausal Data Science Meeting 2021(Talk) Introduction to CausalML at2021 Conference on Digital Experimentation @ MIT (CODE@MIT)(Talk) Causal Inference and Machine Learning in Practice with EconML and CausalML: Industrial Use Cases at Microsoft, TripAdvisor, Uber atKDD 2021 Tutorials(website and slide links)(Publication) CausalML White PaperCausalml: Python package for causal machine learning(Publication)Uplift Modeling for Multiple Treatments with Cost Optimizationat2019 IEEE International Conference on Data Science and Advanced Analytics (DSAA)(Publication)Feature Selection Methods for Uplift ModelingCitationTo cite CausalML in publications, you can refer to the following sources:Whitepaper:CausalML: Python Package for Causal Machine LearningBibtex:@misc{chen2020causalml,
title={CausalML: Python Package for Causal Machine Learning},
author={Huigang Chen and Totte Harinen and Jeong-Yoon Lee and Mike Yung and Zhenyu Zhao},
year={2020},
eprint={2002.11631},
archivePrefix={arXiv},
primaryClass={cs.CY}
}LiteratureChen, Huigang, Totte Harinen, Jeong-Yoon Lee, Mike Yung, and Zhenyu Zhao. "Causalml: Python package for causal machine learning." arXiv preprint arXiv:2002.11631 (2020).Radcliffe, Nicholas J., and Patrick D. Surry. "Real-world uplift modelling with significance-based uplift trees." White Paper TR-2011-1, Stochastic Solutions (2011): 1-33.Zhao, Yan, Xiao Fang, and David Simchi-Levi. "Uplift modeling with multiple treatments and general response types." Proceedings of the 2017 SIAM International Conference on Data Mining. Society for Industrial and Applied Mathematics, 2017.Hansotia, Behram, and Brad Rukstales. "Incremental value modeling." Journal of Interactive Marketing 16.3 (2002): 35-46.Jannik Rößler, Richard Guse, and Detlef Schoder. "The Best of Two Worlds: Using Recent Advances from Uplift Modeling and Heterogeneous Treatment Effects to Optimize Targeting Policies". International Conference on Information Systems (2022)Su, Xiaogang, et al. "Subgroup analysis via recursive partitioning." Journal of Machine Learning Research 10.2 (2009).Su, Xiaogang, et al. "Facilitating score and causal inference trees for large observational studies." Journal of Machine Learning Research 13 (2012): 2955.Athey, Susan, and Guido Imbens. "Recursive partitioning for heterogeneous causal effects." Proceedings of the National Academy of Sciences 113.27 (2016): 7353-7360.Künzel, Sören R., et al. "Metalearners for estimating heterogeneous treatment effects using machine learning." Proceedings of the national academy of sciences 116.10 (2019): 4156-4165.Nie, Xinkun, and Stefan Wager. "Quasi-oracle estimation of heterogeneous treatment effects." arXiv preprint arXiv:1712.04912 (2017).Bang, Heejung, and James M. Robins. "Doubly robust estimation in missing data and causal inference models." Biometrics 61.4 (2005): 962-973.Van Der Laan, Mark J., and Daniel Rubin. "Targeted maximum likelihood learning." The international journal of biostatistics 2.1 (2006).Kennedy, Edward H. "Optimal doubly robust estimation of heterogeneous causal effects." arXiv preprint arXiv:2004.14497 (2020).Louizos, Christos, et al. "Causal effect inference with deep latent-variable models." arXiv preprint arXiv:1705.08821 (2017).Shi, Claudia, David M. Blei, and Victor Veitch. "Adapting neural networks for the estimation of treatment effects." 33rd Conference on Neural Information Processing Systems (NeurIPS 2019), 2019.Zhao, Zhenyu, Yumin Zhang, Totte Harinen, and Mike Yung. "Feature Selection Methods for Uplift Modeling." arXiv preprint arXiv:2005.03447 (2020).Zhao, Zhenyu, and Totte Harinen. "Uplift modeling for multiple treatments with cost optimization." In 2019 IEEE International Conference on Data Science and Advanced Analytics (DSAA), pp. 422-431. IEEE, 2019.Related projectsuplift: uplift models in Rgrf: generalized random forests that include heterogeneous treatment effect estimation in Rrlearner: A R package that implements R-LearnerDoWhy:  Causal inference in Python based on Judea Pearl's do-calculusEconML: A Python package that implements heterogeneous treatment effect estimators from econometrics and machine learning methods