当前位置:主页 > python教程 > python实现最小二乘法线性拟合

python最小二乘法线性拟合实现代码

发布:2020-02-07 17:02:29 181


为网友们分享了python相关的编程文章,网友蒯恺玲根据主题投稿了本篇教程内容,涉及到python、乘法、线性拟合、python实现最小二乘法线性拟合相关内容,已被489网友关注,如果对知识点想更进一步了解可以在下方电子资料中获取。

python实现最小二乘法线性拟合

本文python代码实现的是最小二乘法线性拟合,并且包含自己造的轮子与别人造的轮子的结果比较。

问题:对直线附近的带有噪声的数据进行线性拟合,最终求出w,b的估计值。

最小二乘法基本思想是使得样本方差最小。

代码中self_func()函数为自定义拟合函数,skl_func()为调用scikit-learn中线性模块的函数。

import numpy as np
import matplotlib.pyplot as plt
from sklearn.linear_model import LinearRegression
 
n = 101
 
x = np.linspace(0,10,n)
noise = np.random.randn(n)
y = 2.5 * x + 0.8 + 2.0 * noise
 
def self_func(steps=100, alpha=0.01):
  w = 0.5
  b = 0
  alpha = 0.01
  for i in range(steps):
    y_hat = w*x + b
    dy = 2.0*(y_hat - y)
    dw = dy*x
    db = dy
    w = w - alpha*np.sum(dw)/n
    b = b - alpha*np.sum(db)/n
    e = np.sum((y_hat-y)**2)/n
    #print (i,'W=',w,'\tb=',b,'\te=',e)
  print ('self_func:\tW =',w,'\n\tb =',b)
  plt.scatter(x,y)
  plt.plot(np.arange(0,10,1), w*np.arange(0,10,1) + b, color = 'r', marker = 'o', label = 'self_func(steps='+str(steps)+', alpha='+str(alpha)+')')
 
def skl_func():
  lr = LinearRegression()
  lr.fit(x.reshape(-1,1),y)
  y_hat = lr.predict(np.arange(0,10,0.75).reshape(-1,1))
  print('skl_fun:\tW = %f\n\tb = %f'%(lr.coef_,lr.intercept_))
  plt.plot(np.arange(0,10,0.75), y_hat, color = 'g', marker = 'x', label = 'skl_func')
  
self_func(10000)
skl_func()
plt.legend(loc='upper left')
plt.show()

结果:

self_func:  W = 2.5648753825503197     b = 0.24527830841237772
skl_fun:     W = 2.564875                             b = 0.245278

python实现最小二乘法线性拟合

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持码农之家。


参考资料

相关文章

  • Python3 goto语句的使用实例方法

    发布:2019-12-05

    今天小编就为大家分享一篇对Python3 goto 语句的使用方法详解,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧


  • 怎么查电脑应该安装python哪个版本

    发布:2019-07-03

    在本篇文章里小编给大家分享了关于怎么查电脑应该安装python哪个版本的相关知识点内容,以后需要的朋友们学习参考下。


  • Python人脸识别第三方库face_recognition接口说明文档

    发布:2022-10-26

    给网友朋友们带来一篇关于Python的教程,Python人脸识别第三方库face_recognition接口简单说明,及简单使用方法


  • Python配置文件管理之ini和yaml文件读取的实现

    发布:2023-03-22

    本文主要介绍了Python配置文件管理之ini和yaml文件读取,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧


  • Python+OpenCV实现黑白老照片上色功能

    发布:2023-04-10

    我们都知道,有很多经典的老照片,受限于那个时代的技术,只能以黑白的形式传世。尽管黑白照片别有一番风味,但是彩色照片有时候能给人更强的代入感。本文就来用Python和OpenCV实现老照片上色功能,需要的可以参考一下


  • python射线法检测点是否位于区域外接矩形内检测方法

    发布:2020-04-15

    这篇文章主要为大家详细介绍了python射线法判断检测点是否位于区域外接矩形内,具有一定的参考价值,感兴趣的小伙伴们可以参考一下


  • python编写简易版的贪吃蛇游戏方法

    发布:2019-06-10

    这篇文章主要给大家介绍了关于如何利用python实现简易版的贪吃蛇游戏的相关资料,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着


  • python 生成唯一id的四种方式

    发布:2023-03-21

    本文主要介绍了python 生成唯一id的四种方式,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧


网友讨论