要实现单变量的简单线性回归系数,可以使用 NumPy 中的 'polyfit' 函数。

'polyfit' 函数可以拟合一组数据点,返回拟合的系数。对于简单线性回归,只需要拟合一个一阶多项式即可。

下面是一个示例代码:

import numpy as np

# 输入数据
x = np.array([1, 2, 3, 4, 5])
y = np.array([2, 4, 6, 8, 10])

# 使用 polyfit 进行拟合,返回系数
coefficients = np.polyfit(x, y, 1)

# 打印系数
print(coefficients)

运行上述代码,将得到输出结果:

[2. 0.]

输出结果中的第一个值 2 表示斜率,第二个值 0 表示截距。因此,简单线性回归的方程为 y = 2x + 0。


原文地址: https://www.cveoy.top/t/topic/fviB 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录