import numpy as np
#--------begin---------------
def fit(X, y):
ones = np.ones((X.shape[0], 1))
X = np.concatenate((ones, X), axis=1)
Xt = np.transpose(X)
XtX = np.dot(Xt, X)
XtX_inv = np.linalg.inv(XtX)
XtX_inv_Xt = np.dot(XtX_inv, Xt)
w = np.dot(XtX_inv_Xt, y)
return w.flatten()
#---------end-------------------
if __name__=="__main__":
np.set_printoptions(precision=2)
data = np.array(eval(input()))
X = data[:,:-1]
y = data[:,-1].reshape((-1,1))
print(fit(X,y).flatten())
11111
最新推荐文章于 2025-05-21 23:13:20 发布