python 对于只有一个自变量的线性回归模型,有没有计算k系数和B系数的函数?

ehxuflar  于 2023-02-18  发布在  Python
关注(0)|答案(1)|浏览(122)

我知道我可以自己编写所需的方法,但是必须有一个函数来解决这个问题,因为这个问题太常见了,如果有人不明白我在说什么,看看下面的公式{Image must be here}
例如,我有一个函数y = kx+B,其中y是因变量,x是自变量,我需要计算k(斜率)和B(截距),我从图中得到了公式,以及那些公式所需要的一切。在常见的数据科学库中有没有什么函数可以帮助计算那些公式?我提到过“只有一个自变量”因为有时有多个独立的变量,这导致多维图
我已经使用了自己的实现,但是我更喜欢scipy和numpy或者sklearn等包中的原生函数

olhwl3o2

olhwl3o21#

我没有把握完全理解这个问题(特别是,你说的“一个自变量”是什么意思?),所以我试着重新表述。如果你有两个变量,xy,都用样本(x_1,..., x_n), (y_1,..., y_n)表示,并且你怀疑它们之间存在线性关系,y = a*x +b,然后你可以使用numpy.polyfit来求系数ab

import numpy as np

n = 100
x = np.linspace(0, 1, n)
y = 2*x + 0.3

a, b = np.polyfit(x, y, 1)
print(f"a={a}, b={b}")

返回

a=2.0, b=0.30000000000000016

希望能有所帮助!

相关问题