当p> n时,sklearn如何做线性回归?

时间:2014-05-17 18:04:46

标签: scikit-learn regression

已知当变量数(p)大于样本数(n)时,未定义最小二乘估计量。

在sklearn中,我收到了这个值:

In [30]: lm = LinearRegression().fit(xx,y_train)

In [31]: lm.coef_
Out[31]: 
array([[ 0.20092363, -0.14378298, -0.33504391, ..., -0.40695124,
         0.08619906, -0.08108713]])

In [32]: xx.shape
Out[32]: (1097, 3419)

调用[30]应该返回错误。在这种情况下,当p> n时,sklearn如何工作?

编辑: 似乎矩阵充满了一些值

if n > m:
        # need to extend b matrix as it will be filled with
        # a larger solution matrix
        if len(b1.shape) == 2:
            b2 = np.zeros((n, nrhs), dtype=gelss.dtype)
            b2[:m,:] = b1
        else:
            b2 = np.zeros(n, dtype=gelss.dtype)
            b2[:m] = b1
        b1 = b2

1 个答案:

答案 0 :(得分:5)

当线性系统不确定时,sklearn.linear_model.LinearRegression找到最小L2范数解,即

argmin_w l2_norm(w) subject to Xw = y

通过将X的伪逆应用于y,即

,它总是很明确并且可以获得。
w = np.linalg.pinv(X).dot(y)

scipy.linalg.lstsq使用的LinearRegression的具体实现使用get_lapack_funcs(('gelss',), ...,它正是通过奇异值分解(由LAPACK提供)找到最小范数解的求解器。

查看此示例

import numpy as np
rng = np.random.RandomState(42)
X = rng.randn(5, 10)
y = rng.randn(5)

from sklearn.linear_model import LinearRegression
lr = LinearRegression(fit_intercept=False)
coef1 = lr.fit(X, y).coef_
coef2 = np.linalg.pinv(X).dot(y)

print(coef1)
print(coef2)

你会看到coef1 == coef2。 (请注意,fit_intercept=False是在sklearn估算器的构造函数中指定的,否则它会在拟合模型之前减去每个要素的平均值,从而产生不同的系数。)