怎么用sklearn学习简单的线性回归?

 时间:2026-02-14 02:06:16

1、请安装anaconda的python3+,这里面已经自带了需要用到的模块,比如sklearn、scipy等模块。

怎么用sklearn学习简单的线性回归?

2、这里要处理的问题是——

给定一组数据X和y

X = [[0,0],[1,1],[2,2]]

y = [0,1,2]

需要拟合一个二元函数f,使得:

f(X[i])=y[i],其中,i=0,1,2

怎么用sklearn学习简单的线性回归?

3、这里介绍一下用线性表达式来表示这个f,需要加载下面的模块:

from sklearn import linear_model

怎么用sklearn学习简单的线性回归?

4、创建一个有待定系数的线性函数:

f = linear_model.LinearRegression()

怎么用sklearn学习简单的线性回归?

5、准备数据:

X = [[0,0],[1,1],[2,2]]

y = [0,1,2]

怎么用sklearn学习简单的线性回归?

6、拟合f,得到f的待定系数:

f.fit(X,y)

print(f.coef_)

怎么用sklearn学习简单的线性回归?

7、知道了系数,就可以构造这个函数:

import numpy as np

a = f.coef_

def f(n):

    b = np.dot(np.array(n),a)

    return b

用这个函数来预测:

print(f([6,6]))

怎么用sklearn学习简单的线性回归?

8、f得到的预测结果是64位的浮点数:

print(f([5,5]).dtype)

怎么用sklearn学习简单的线性回归?

9、所以,这里需要在函数f的return的数据转一下数据类型:

def f(n):

    b = np.dot(np.array(n),a)

    return round(b).astype(np.int8)

四舍五入取整,再转化为int8类型的数据。

怎么用sklearn学习简单的线性回归?

  • 火影忍者ol忍者新时代怎么召唤忍具
  • 如何下载简单搜索答题神器
  • 四种人最容易成功,你是其中之一吗?
  • 西南大学心理学
  • 燕窝酒酿蛋冲泡方法 效果更好哦
  • 热门搜索
    pte是什么 腺体是什么 jk制服是什么梗 设区市是什么意思 郁金香什么时候开花 苹果id是什么 流标是什么意思 麻豆是什么意思 莆田系医院是什么意思 巨蟹座和什么星座最配