#P4484. 线性回归(最小二乘法)

线性回归(最小二乘法)

题目描述

实现一元线性回归,使用最小二乘法求解 y=wx+by = wx + b。 给定训练样本点集合 points,其中每个元素为 [x, y],请计算并返回拟合得到的参数 wwbb

输入参数

  • points:二维列表,每个元素为长度为 2 的浮点数列表 [x, y],表示一个训练样本点

返回值

  • [w, b]:长度为 2 的列表,分别为拟合直线的斜率 ww 和截距 bb(可按浮点数形式输出)

示例 1:

输入:

points = [[1.0, 2.0],[2.0, 4.0],[3.0, 6.0],[4.0, 8.0],[5.0, 10.0]]

输出:

[w, b] = [2.0000, 0.0000]

提示

  • 1<=len(points)<=51041 <= len(points) <= 5*10^4
  • 105<=x,y<=105-10^5 <= x, y <= 10^5
  • 至少存在两个不同的 x 值。