使用sm.OLS时需要添加常量吗？

Do I need to add a constant when using sm.OLS?(使用sm.OLS时需要添加常量吗？)

问题描述

我正在对两组数据Y和X执行OLS。我使用statsmod.api.OLS。然而，我发现无论我之前是否向X添加一个常量，结果都非常不同。代码如下：

import statsmodels.api as sm
import numpy as np

mess = "SELECT .... FROM... WHERE ...."
data = np.array(db.extractData(mess))
Y = data[,:0]
X = data[,:1]
#Option1 
res = sm.OLS(Y,X).fit().rsquared ---> will return 0.76
#Option2
X = sm.add_constant(X)
res = sm.OLS(Y,X).fit().rsquared ---> will return 0.06

考虑到我是否添加常量的巨大差异，我假设我做错了什么。非常感谢您抽出时间来。

推荐答案

您需要添加常量。来自文档：http://www.statsmodels.org/devel/generated/statsmodels.regression.linear_model.OLS.html

An intercept is not included by default and should be added by the user. See statsmodels.tools.add_constant.

这篇关于使用sm.OLS时需要添加常量吗？的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持编程学习网！