causalml

causalml

Python因果推断与提升建模库:causalml

causalml是一个Python库,集成了机器学习算法用于提升建模和因果推断。它提供标准接口,支持从实验或观察数据中估计条件平均处理效应和个体处理效应。该库适用于广告定向优化和个性化推荐等场景,有助于提高营销效果。causalml实现了多种因果推断方法,并配有详细文档和示例,便于开发者学习和应用。

CausalML因果推断机器学习个性化推荐营销优化Github开源项目
<div align="center"> <a href="https://github.com/uber/causalml"><img width="380px" height="140px" src="https://yellow-cdn.veclightyear.com/2b54e442/f0aa0527-f542-4640-8445-072892039425.png"></a> </div>

PyPI 版本 构建状态 文档状态 下载量 CII 最佳实践

免责声明

该项目稳定并正在孵化以获得长期支持。它可能包含新的实验性代码,其API可能会发生变化。

Causal ML:用于提升建模和因果推断的Python包

Causal ML是一个Python包,它提供了一套基于最新研究的使用机器学习算法进行提升建模和因果推断的方法[1]。它提供了一个标准接口,允许用户从实验数据或观察数据中估计条件平均处理效应(CATE)或个体处理效应(ITE)。本质上,它估计干预T对具有观察特征X的用户的结果Y的因果影响,而不对模型形式做出强假设。典型的使用场景包括

  • 营销活动目标优化:提高广告活动投资回报率的一个重要杠杆是将广告定向投放给在特定KPI(如参与度或销售额)方面会有积极响应的客户群。CATE通过从A/B实验或历史观察数据中估计个体层面的广告曝光对KPI的影响来识别这些客户。

  • 个性化互动:公司有多种与客户互动的选择,如追加销售的不同产品选择或沟通的消息渠道。可以使用CATE来估计每个客户和处理选项组合的异质处理效应,从而建立最佳的个性化推荐系统。

文档

文档可在以下地址获取:

https://causalml.readthedocs.io/en/latest/about.html

安装

安装说明可在以下地址获取:

https://causalml.readthedocs.io/en/latest/installation.html

快速入门

包含代码片段的快速入门指南可在以下地址获取:

https://causalml.readthedocs.io/en/latest/quickstart.html

示例笔记本

示例笔记本可在以下地址获取:

https://causalml.readthedocs.io/en/latest/examples.html

贡献

我们欢迎社区贡献者参与项目。在开始之前,请阅读我们的行为准则并查看贡献指南

版本控制

我们在更新日志中记录版本和变更。

许可证

本项目采用Apache 2.0许可证 - 详情请参阅LICENSE文件。

参考文献

文档

CausalML团队的会议演讲和出版物

引用

要在出版物中引用CausalML,您可以参考以下来源:

白皮书: CausalML: Python Package for Causal Machine Learning

Bibtex:

@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} }

文献

  1. Chen, Huigang, Totte Harinen, Jeong-Yoon Lee, Mike Yung, and Zhenyu Zhao. "Causalml: Python package for causal machine learning." arXiv preprint arXiv:2002.11631 (2020).
  2. 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.
  3. 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.
  4. Hansotia, Behram, and Brad Rukstales. "Incremental value modeling." Journal of Interactive Marketing 16.3 (2002): 35-46.
  5. 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)
  6. Su, Xiaogang, et al. "Subgroup analysis via recursive partitioning." Journal of Machine Learning Research 10.2 (2009).
  7. Su, Xiaogang, et al. "Facilitating score and causal inference trees for large observational studies." Journal of Machine Learning Research 13 (2012): 2955.
  8. Athey, Susan, and Guido Imbens. "Recursive partitioning for heterogeneous causal effects." Proceedings of the National Academy of Sciences 113.27 (2016): 7353-7360.
  9. 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.
  10. Nie, Xinkun, and Stefan Wager. "Quasi-oracle estimation of heterogeneous treatment effects." arXiv preprint arXiv:1712.04912 (2017).
  11. Bang, Heejung, and James M. Robins. "Doubly robust estimation in missing data and causal inference models." Biometrics 61.4 (2005): 962-973.
  12. Van Der Laan, Mark J., and Daniel Rubin. "Targeted maximum likelihood learning." The international journal of biostatistics 2.1 (2006).
  13. Kennedy, Edward H. "Optimal doubly robust estimation of heterogeneous causal effects." arXiv preprint arXiv:2004.14497 (2020).
  14. Louizos, Christos, et al. "Causal effect inference with deep latent-variable models." arXiv preprint arXiv:1705.08821 (2017).
  15. 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.
  16. Zhao, Zhenyu, Yumin Zhang, Totte Harinen, and Mike Yung. "Feature Selection Methods for Uplift Modeling." arXiv preprint arXiv:2005.03447 (2020).
  17. 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.

相关项目

  • uplift:R中的提升模型
  • grf:R中包括异质处理效应估计的广义随机森林
  • rlearner:实现R-Learner的R包
  • DoWhy:基于Judea Pearl的do-calculus的Python因果推断
  • EconML:实现来自计量经济学和机器学习方法的异质处理效应估计器的Python包

编辑推荐精选

QoderWork

QoderWork

阿里Qoder团队推出的桌面端AI智能体

QoderWork 是阿里推出的本地优先桌面 AI 智能体,适配 macOS14+/Windows10+,以自然语言交互实现文件管理、数据分析、AI 视觉生成、浏览器自动化等办公任务,自主拆解执行复杂工作流,数据本地运行零上传,技能市场可无限扩展,是高效的 Agentic 生产力办公助手。

音述AI

音述AI

全球首个AI音乐社区

音述AI是全球首个AI音乐社区,致力让每个人都能用音乐表达自我。音述AI提供零门槛AI创作工具,独创GETI法则帮助用户精准定义音乐风格,AI润色功能支持自动优化作品质感。音述AI支持交流讨论、二次创作与价值变现。针对中文用户的语言习惯与文化背景进行专门优化,支持国风融合、C-pop等本土音乐标签,让技术更好地承载人文表达。

lynote.ai

lynote.ai

一站式搞定所有学习需求

不再被海量信息淹没,开始真正理解知识。Lynote 可摘要 YouTube 视频、PDF、文章等内容。即时创建笔记,检测 AI 内容并下载资料,将您的学习效率提升 10 倍。

AniShort

AniShort

为AI短剧协作而生

专为AI短剧协作而生的AniShort正式发布,深度重构AI短剧全流程生产模式,整合创意策划、制作执行、实时协作、在线审片、资产复用等全链路功能,独创无限画布、双轨并行工业化工作流与Ani智能体助手,集成多款主流AI大模型,破解素材零散、版本混乱、沟通低效等行业痛点,助力3人团队效率提升800%,打造标准化、可追溯的AI短剧量产体系,是AI短剧团队协同创作、提升制作效率的核心工具。

seedancetwo2.0

seedancetwo2.0

能听懂你表达的视频模型

Seedance two是基于seedance2.0的中国大模型,支持图像、视频、音频、文本四种模态输入,表达方式更丰富,生成也更可控。

nano-banana纳米香蕉中文站

nano-banana纳米香蕉中文站

国内直接访问,限时3折

输入简单文字,生成想要的图片,纳米香蕉中文站基于 Google 模型的 AI 图片生成网站,支持文字生图、图生图。官网价格限时3折活动

扣子-AI办公

扣子-AI办公

职场AI,就用扣子

AI办公助手,复杂任务高效处理。办公效率低?扣子空间AI助手支持播客生成、PPT制作、网页开发及报告写作,覆盖科研、商业、舆情等领域的专家Agent 7x24小时响应,生活工作无缝切换,提升50%效率!

堆友

堆友

多风格AI绘画神器

堆友平台由阿里巴巴设计团队创建,作为一款AI驱动的设计工具,专为设计师提供一站式增长服务。功能覆盖海量3D素材、AI绘画、实时渲染以及专业抠图,显著提升设计品质和效率。平台不仅提供工具,还是一个促进创意交流和个人发展的空间,界面友好,适合所有级别的设计师和创意工作者。

图像生成AI工具AI反应堆AI工具箱AI绘画GOAI艺术字堆友相机AI图像热门
码上飞

码上飞

零代码AI应用开发平台

零代码AI应用开发平台,用户只需一句话简单描述需求,AI能自动生成小程序、APP或H5网页应用,无需编写代码。

Vora

Vora

免费创建高清无水印Sora视频

Vora是一个免费创建高清无水印Sora视频的AI工具

下拉加载更多