R lm与Python sklearn linear_model

时间:2016-10-11 01:06:23

标签: python r scikit-learn regression linear

当我研究Python SKlearn时,我遇到的第一个例子是Generalized Linear Models

第一个例子的代码:

from sklearn import linear_model
reg = linear_model.LinearRegression()
reg.fit([[0, 0], [1, 1], [2,2]], [0, 1,2])
reg.fit
reg.coef_
array([ 0.5,  0.5])

我假设[[0, 0], [1, 1], [2,2]]代表的数据框也包含x1 = c(0,1,2)x2 = c(0,1,2)以及y = c(0,1,2)

我立刻开始认为array([ 0.5, 0.5])x1x2的系数。

但是,这些估算有标准错误吗?那么t测试p值,R2和其他数据呢?

然后我尝试在R中做同样的事情。

X = data.frame(x1 = c(0,1,2),x2 = c(0,1,2),y = c(0,1,2))
lm(data=X, y~x1+x2)
Call:
lm(formula = y ~ x1 + x2, data = X)

#Coefficients:
#(Intercept)           x1           x2  
#  1.282e-16    1.000e+00           NA  

显然x1x2完全线性相关,因此OLS会失败。为什么SKlearn仍然有效并给出了这个结果?我是以错误的方式得到了sklearn吗?感谢。

1 个答案:

答案 0 :(得分:6)

两种解决方案都是正确的(假设NA表现得像零)。哪种解决方案更受青睐取决于OLS估算器使用的数值求解器。

sklearn.linear_model.LinearRegression基于scipy.linalg.lstsq,后者又调用此处描述的LAPACK gelsd例程:

http://www.netlib.org/lapack/lug/node27.html

特别是它说当问题排名不足时,它会寻求最小范数最小二乘解。

如果你想支持另一个解决方案,你可以使用一个坐标下降求解器,在Lasso类中实现一点点L1惩罚:

>>> from sklearn.linear_model import Lasso
>>> reg = Lasso(alpha=1e-8)
>>> reg.fit([[0, 0], [1, 1], [2, 2]], [0, 1, 2])

Lasso(alpha=1e-08, copy_X=True, fit_intercept=True, max_iter=1000,
   normalize=False, positive=False, precompute=False, random_state=None,
   selection='cyclic', tol=0.0001, warm_start=False)
>>> reg.coef_
array([  9.99999985e-01,   3.97204719e-17])