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

2024-10-15 03:10:52

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

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

3、这里介绍一下用线性表嚼秒躞句达式来表示这个f,需要加载下面的模块:from sklearn import linear_model

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

5、准备数据:X = [[0,0],[1,1],[2,2]]y = [0,1,2]

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

7、知道了系数,就可以构造这个函数:import numpy as npa = f.coef_def f(n): b = np.dot(np.array(n),a) return b用这个函数来预测:print(f([6,6]))

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

9、所以,这里需要在函数f的return的数据转一下数据类型:def f(n): b = np.dot(np.array(n),a) return round(b).ast烤恤鹇灭ype(np.int8)四舍五入取整,再转化为int8类型的数据。

怎么用sklearn学习简单的线性回归
猜你喜欢