案例：使用pytorch实现一元线性回归模型

                                                                                      点击此处返回总目录

 

一、什么是线性回归

线性回归是分析一个变量与另一个变量的关系的方法。

线性回归的意思就是指x与y之间的关系是线性的。y = wx+b

 

二、求解的步骤

1. 确定模型。                 这里选择y = wx+b

2. 选择损失函数。          这里采用均方差误差。

3. 求梯度并更新w和b。  w = w - l * w.grad         b = b - l * b.grad

 

三、如果不使用pytorch，如何实现线性回归

'''
python实现一元线性回归：y = wx + b
使用的是批量梯度下降算法（所有样本算一次梯度）
'''

import numpy as np
import matplotlib.pyplot as plt

learning_rate = 0.01  #学习率

#创建训练数据
#np.random.seed(123)
x_train = np.random.rand(20) * 10