开源仓库py-why/EconML4,708 Stars

EconML

异质性处理效应估计库,用正交机器学习估计个体化因果效应

适合需要估计处理对不同个体差异化影响的研究者和数据科学家。它把任意sklearn模型用作干扰和最终阶段,通过交叉拟合去偏,给出带置信区间的个体化因果效应、策略推荐和SHAP解释。

bo老师判断:推荐

用途和边界相对清楚,可以先从小样本开始。

...互动状态已更新

这个工具提供了哪些能力

点开任一能力,看清怎么用

每个能力展开后是三件事:它会给你什么、你需要准备什么、哪些必须由你把关。

常见场景:想估计处理对不同人群的差异化影响而不仅是平均效应;观察数据中存在大量控制变量需要非参数控制;希望得到个体化处理推荐(决策树/森林策略);需要比较多个CATE模型的样本外表现并选择最佳模型

覆盖阶段设计分析

第一次这样开始

把仓库地址直接交给 AI

先让 AI 说明环境和风险,再完成最小安装验证。开始前想清楚要用哪个能力、把对应材料备在手边——每张能力卡展开后有具体清单。

第一次直接复制

帮我安装这个库:https://github.com/py-why/EconML
安装前先告诉我需要什么环境,安装后帮我跑通一个最小示例。

别这样用

这些判断,仍然由你负责

  • 在离散处理场景忘记设置discrete_treatment=True
  • 用元学习器时未把混淆变量W拼入特征矩阵X
  • 把SHAP值解释为因果归因而非对CATE预测的贡献度量
  • 在无有效工具变量的场景强行使用IV估计器

技术依据

查看能力拆解、验证范围与来源

处理机制

  1. 01

    核心处理链

    异质性处理效应估计库,用正交机器学习估计个体化因果效应

已核对范围

  • 静态读取 dml/dml.py DML/LinearDML/SparseLinearDML/KernelDML/NonParamDML/BaseDML 类继承与 fit 构造
  • 静态读取 dml/rlearner.py RLearner/ModelNuisance/ModelFinal 机制
  • 静态读取 dml/causalforest.py CausalForestDML 继承与 BLB inference
  • 静态读取 metalearners/metalearners.py SLearner/TLearner/XLearner/DomainAdaptationLearner 全部类
  • 静态读取各类的 model 构造、fit、effect 与 bootstrap inference
  • 静态读取 dr/drlearner.py DRLearner 及 ModelNuisance/ModelFinal/BaseDR
  • 静态读取 DRLearner estimating equations(propensity + regression nuisance)
  • 静态读取 LinearDRLearner/SparseLinearDRLearner/ForestDRLearner 的 final-stage mixin

仍待核对

  • 未运行任一 DML 估计器,未验证 CATE、CI、coef 实际数值
  • 未验证 final-stage 选择对推断的影响
  • 未运行任一元学习器,未验证 CATE、bootstrap CI 实际数值
  • 未验证 DomainAdaptationLearner 的倾向性加权 final 行为
  • 未运行任一 DR 估计器,未验证 CATE、CI、coef 实际数值
  • 未验证多处理臂(非 baseline)的 final multi-task 行为
  • 未运行任一 IV 估计器,未验证 CATE、CI 实际数值
  • 未验证 IntentToTreat(二元 Z、二元 T)的特殊 final 路径

原仓库

py-why/EconML

读取版本

e546416862280c90f9b21e95c298d00d599c75c8

许可

MIT

最近核对

内容 2026-07-10
Stars 2026-07-11

部分完成验证

科研之我见

AI 科研全流程怎么搭,我梳理出三个核心入口

每天开工先固定阶段入口、材料入口和验证入口,把庞大的科研全流程收缩成一段可执行、可检查的任务。

查看内容

科研之我见

AI 可以参与科研的哪些环节?一张 11 环节工作流全景图

从选题、文献、理论、数据到投稿,逐环节判断 AI 可以主导、需要人工校验和必须由研究者把关的工作。

查看内容

文献 · 分析 · 写作

grobid

把学术论文PDF解析成结构化TEI XML,提取元数据、全文结构和参考文献

查看内容

设计 · 分析 · 验证

CausalPy

用贝叶斯方法做准实验因果推断,含10+方法和稳健性诊断套件

查看内容

文献 · 设计 · 分析 · 写作

Scientific-Agent-Skills

为AI科研代理提供文献检索、写作、统计、实验设计和同行评审的结构化能力。

查看内容

相关课程

把单项工具放进完整研究设计流程

课程一从研究问题、文献判断、理论假设和方法骨架继续推进,帮助你建立可以复用的研究设计档案。

查看课程一