一元线性回归模型 文章目录 一元线性回归模型 一、求解步骤 二、代码实现 三、实现效果 一、求解步骤 1.确定模型 Model: y=wx+b 2.选择损失函数 MES 3.求解梯度并更新w,b w = w - LR * w.grade b = b - LR * b.grade 二、代码实现 #一元线性回归模型 import torch import matplotlib.pyplot as plt torch.manual_seed(10) lr=0.05 #学习率 #创建训练数据 x = torch.rand(20,1)*10 y = 2*x+(5+torch.randn(20,1))#加上随机噪声 #构建线性回归参数