# 线性回归问题的梯度下降算法
import numpy as np
class LinearRegression:
def fit(self,X,y,eta,N):
m,n=X.shape
w=np.zeros((n,1))
for t in range(N):
e=X.dot(w)-y
g=2*X.T.dot(e)/m
w=w-eta*g
self.w=w
def predict(self,X):
return X.dot(self.w)
机器学习算法导论代码---linear_regression_gd
最新推荐文章于 2023-09-07 17:13:21 发布