#P4484. 线性回归(最小二乘法)
线性回归(最小二乘法)
题目描述
实现一元线性回归,使用最小二乘法求解 。
给定训练样本点集合 points,其中每个元素为 [x, y],请计算并返回拟合得到的参数 和 。
输入参数
points:二维列表,每个元素为长度为 2 的浮点数列表[x, y],表示一个训练样本点
返回值
[w, b]:长度为 2 的列表,分别为拟合直线的斜率 和截距 (可按浮点数形式输出)
示例 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]
提示
- 至少存在两个不同的
x值。