编写一个使用梯度下降执行线性回归的 Python 函数。该函数应将 NumPy 数组 X(具有一列截距的特征)和 y(目标)作为输入,以及学习率 alpha 和迭代次数,并返回一个 NumPy 数组,表示线性回归模型的系数。
输入描述:
第1行输入X,第2行输入y,第3行输入alpha,第4行输入迭代次数。


输出描述:
输出线性回归模型的系数,四舍五入到小数点后四位。返回类型是List类型。
示例1

输入

[[1, 1], [1, 2], [1, 3], [1, 4]]
[2, 3, 4, 5]
0.01
1000

输出

[0.8678 1.045 ]

备注:
1.对应的输入、输出已给出,您只用实现核心功能函数即可。2.支持numpy、scipy、pandas、scikit-learn库。
加载中...