100字范文,内容丰富有趣,生活中的好帮手!
100字范文 > 使用线性回归拟合平面最佳直线及预测之Python+sklearn实现

使用线性回归拟合平面最佳直线及预测之Python+sklearn实现

时间:2020-06-08 03:43:22

相关推荐

使用线性回归拟合平面最佳直线及预测之Python+sklearn实现

本文代码采用sklearn扩展库实现,使用线性回归算法解决下面的问题:根据平面上已知3个点的坐标,拟合最佳直线斜率k和截距b,然后根据拟合的结果对给出的x坐标进行预测,得到y坐标。

fromsklearnimportlinear_model

deflinearRegressionPredict(x, y):

lr = linear_model.LinearRegression()

# 拟合

lr.fit(x, y)

returnlr

# 平面上三个点的x轴坐标

x = [[1], [5], [7]]

# 平面上三个点的y轴坐标

y = [[3], [100], [120]]

# 根据已知3个点拟合最佳直线的系数和截距

lr = linearRegressionPredict(x, y)

# 查看最佳拟合系数

print('k:', lr.coef_)

# 截距

print('b:', lr.intercept_)

# 测试代码,预测

xs = [[[3]], [[5]], [[7]], [[10]]]

foriteminxs:

print(item,':', lr.predict(item))

运行结果:

k: [[ 20.17857143]]

b: [-13.10714286]

[[3]] : [[ 47.42857143]]

[[5]] : [[ 87.78571429]]

[[7]] : [[ 128.14285714]]

[[10]] : [[ 188.67857143]]

----------喜大普奔----------

1、继《Python程序设计基础》《Python程序设计(第2版)》《Python可以这样学》系列图书之后,董付国老师新书《Python程序设计开发宝典》已于8月1日在清华大学出版社出版。为庆祝新书《Python程序设计开发宝典》全面上架,清华大学出版社联合“赣江图书专营”淘宝店推出特价优惠活动,《Python程序设计开发宝典》原价69元,新书上架期间超低价39.8元,可以复制下面的链接使用浏览器打开查看图书详情和购买:

/item.htm?spm=a1z10.3-b-s.w4011-14464369246.84.46f16db0roWfX4&id=557107249812&rn=339cbc9df2bac424664103917dedfbd2&abbucket=8&tbpm=3

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。