DoubleML：Python中的双重机器学习框架

2026-02-06 05:06:49作者：齐冠琰

DoubleML是一个基于Python的开源库，实现了Chernozhukov等人(2018)提出的双重/去偏机器学习框架。该库设计用于部分线性回归模型(PLR)、部分线性工具变量回归模型(PLIV)、交互式回归模型(IRM)以及交互式工具变量回归模型(IIVM)。

项目特点

DoubleML基于scikit-learn构建，具有灵活的对象导向结构，支持高度定制化的模型设定。主要功能包括：

部分线性回归模型(PLR)估计
部分线性工具变量回归模型(PLIV)估计
交互式回归模型(IRM)估计
交互式工具变量回归模型(IIVM)估计
灵活的机器学习方法选择
完整的估计与统计推断流程

环境要求与安装

系统要求

Python 3.9或更高版本
依赖库：numpy、pandas、scipy、scikit-learn、statsmodels、joblib、matplotlib、seaborn、plotly

安装方法

方法一：使用pip安装

pip install -U DoubleML

方法二：从源代码安装

git clone https://gitcode.com/gh_mirrors/do/doubleml-for-py.git
cd doubleml-for-py
pip install --editable .

快速开始示例

以下是一个使用DoubleML进行部分线性回归模型估计的基本示例：

from doubleml import DoubleMLData, DoubleMLPLR
from sklearn.linear_model import Lasso
import pandas as pd
import numpy as np

# 创建示例数据
n_obs = 500
n_vars = 100
X = np.random.normal(size=(n_obs, n_vars))
d = np.dot(X[:, :3], [1, 2, 3]) + np.random.normal(size=n_obs)
y = d + np.dot(X[:, :3], [1, 2, 3]) + np.random.normal(size=n_obs)

df = pd.DataFrame(X, columns=[f'X{i}' for i in range(n_vars)])
df['y'] = y
df['d'] = d

# 创建DoubleML数据对象
dmldata = DoubleMLData.from_pandas(df, y_col='y', treatment_col='d', 
                                  covar_col=[f'X{i}' for i in range(n_vars)])

# 初始化Lasso回归器
lasso = Lasso(alpha=0.1)

# 创建PLR模型
dml_plr = DoubleMLPLR(dmldata, lasso, lasso)

# 拟合模型
dml_plr.fit()

# 输出结果
print("系数估计:", dml_plr.coef)
print("标准误差:", dml_plr.se)
print("p值:", dml_plr.pval)
print("95%置信区间:", dml_plr.confint())