实现简单的线性回归
【案例】给定一组数据(1, 1) , (2, 3) , (3, 2) , (4, 3) , (5, 5)。试通过简单线性回归求出拟合的直线,并能根据拟合的直线计算给定数据集的预测值。
(1)绘制出图像。
(2)根据公式计算a和b的值。
(3)将要预测的数据代入到拟合直线的方程进行计算。
红色直线即我们通过简单线性回归得到的拟合函数。
将简单线性回归编写为一个类,在类中包含初始化回归模型,训练回归模型和预测等方法。
(1)初始化简单线性回归模型
【案例】给定一组数据(1, 1) , (2, 3) , (3, 2) , (4, 3) , (5, 5)。试通过简单线性回归求出拟合的直线,并能根据拟合的直线计算给定数据集的预测值。
(1)绘制出图像。
(2)根据公式计算a和b的值。
(3)将要预测的数据代入到拟合直线的方程进行计算。
红色直线即我们通过简单线性回归得到的拟合函数。
将简单线性回归编写为一个类,在类中包含初始化回归模型,训练回归模型和预测等方法。
(1)初始化简单线性回归模型