天天看點

最小二乘法與多元線性回歸

設有方程組Xβ=Y,求解β

設y=2*x0 + 3*x1 + 1

 當X = [ [1, 1, 1],

               [1, 2, 1],

               [2, 1, 1]]時

  β=[ [x0], [x1], [x2] ]

       Y=[ [6], [9], [8] ]

用numpy求解:

from numpy import linalg import lstsq
print lstsq(X, Y)[0]#輸出的為β
           

用sklearn:

model = linearRegression()
model.fit(X, Y)
x2 = [ [1,2,3] ]
y2 = model.predict(x2)
           

繼續閱讀