在 statsmodels 上使用样本权重进行最小二乘回归

Question

我希望在统计模型上使用样本权重实施 OLS。具体应用是美国时间使用调查，其中样本权重根据人口的人口平衡进行调整。如果没有直接实现，那么协助使用样本权重对估计器进行硬编码也会有所帮助。

Answer 1

@Mario，你可以试试 WLS 来自 statsmodels

文档中的示例。

import numpy as np
import statsmodels.api as sm

Y = [1,3,4,5,2,3,4]
X = range(1,8)
X = sm.add_constant(X)
wls_model = sm.WLS(Y,X, weights=list(range(1,8)))
results = wls_model.fit()

print (results.params)

更多示例here。