python计算一元线性回归拟合方程

最新推荐文章于 2024-08-09 16:21:54 发布

Memory_and_Dream

最新推荐文章于 2024-08-09 16:21:54 发布

阅读量3.7k

点赞数 4

本文链接：https://blog.csdn.net/Memory_and_Dream/article/details/105841127

版权

# 导入sklearn库下的linear_model类
from sklearn import linear_model

# 一元线性回归
# 转化自变量数据为矩阵
x = [[6], [8], [10], [14], [18]]
y = [[7], [9], [13], [17.5], [18]]  # 现在只能用二元矩阵
# print(x)
# print(y)

# 调用线性回归函数
clf = linear_model.LinearRegression()
# 开始线性回归计算
clf.fit(x,y)
# 得到斜率
print(clf.coef_[0])
# 得到截距
print(clf.intercept_)