在岭回归中,我们正在使用Ax=b
正则化来解决L2
。直接计算如下:
x =(A T A + alpha * I) -1 A T b
我查看了scikit-learn代码,他们确实执行了相同的计算。但是,我似乎无法为alpha > 0
重现此内容的最小代码。
import numpy as np
A = np.asmatrix(np.c_[np.ones((10,1)),np.random.rand(10,3)])
b = np.asmatrix(np.random.rand(10,1))
I = np.identity(A.shape[1])
alpha = 1
x = np.linalg.inv(A.T*A + alpha * I)*A.T*b
print(x.T)
>>> [[ 0.37371021 0.19558433 0.06065241 0.17030177]]
from sklearn.linear_model import Ridge
model = Ridge(alpha = alpha).fit(A[:,1:],b)
print(np.c_[model.intercept_, model.coef_])
>>> [[ 0.61241566 0.02727579 -0.06363385 0.05303027]]
有关如何解决这种差异的建议吗?
答案 0 :(得分:2)
对于直接版本和numpy版本,此修改似乎产生相同的结果:
import numpy as np
A = np.asmatrix(np.random.rand(10,3))
b = np.asmatrix(np.random.rand(10,1))
I = np.identity(A.shape[1])
alpha = 1
x = np.linalg.inv(A.T*A + alpha * I)*A.T*b
print (x.T)
from sklearn.linear_model import Ridge
model = Ridge(alpha = alpha, tol=0.1, fit_intercept=False).fit(A ,b)
print model.coef_
print model.intercept_
似乎差异的主要原因是,班级Ridge
具有参数fit_intercept=True
(通过继承自班级_BaseRidge
)(source)
这是在将矩阵传递给_solve_cholesky
函数之前应用数据居中过程。
这是Ridge.py中的行
X, y, X_mean, y_mean, X_std = self._center_data(
X, y, self.fit_intercept, self.normalize, self.copy_X,
sample_weight=sample_weight)
此外,您似乎试图通过添加1的列来隐含地考虑拦截。如您所见,如果您指定fit_intercept=False
附录:Ridge类实际上是否实现了直接公式?
这取决于solver
参数的选择。
实际上,如果您未在solver
中指定Ridge
参数,则默认情况下会solver='auto'
(内部转换为solver='cholesky'
)。这应该等同于直接计算。
严格来说,_solve_cholesky
使用numpy.linalg.solve
代替numpy.inv
。但可以很容易地检查
np.linalg.solve(A.T*A + alpha * I, A.T*b)
与
相同np.linalg.inv(A.T*A + alpha * I)*A.T*b