梯度下降法求解w和b
导入所需要的包:
#导入sklearn的线性回归算法只是为了验证手写的回归算法w和b求解是否正确
import numpy as np
import matplotlib.pyplot as plt
%matplotlib inline
from sklearn.linear_model import LinearRegression
创建数据集:
X = np.linspace(2.5,12,25)
w = np.random.randint(2,10,size = 1)[0]
b = np.random.randint(-5,5,size = 1)[0]
y = X*w + b + np.random.randn(25)*2
数据点:
梯度下降法实现回归:
class Linear_model