Python 用5行代码学机器学习—线性回归

开发 后端
我准备使用scikit-learn给大家介绍一些模型的基础知识,今天就来讲讲线性回归模型。

我准备使用scikit-learn给大家介绍一些模型的基础知识,今天就来讲讲线性回归模型。

1. 准备

开始之前,你要确保Python和pip已经成功安装在电脑上噢,如果没有,请访问这篇文章:超详细Python安装指南 进行安装。如果你用Python的目的是数据分析,可以直接安装Anaconda:Python数据分析与挖掘好帮手—Anaconda

Windows环境下打开Cmd(开始—运行—CMD),苹果系统环境下请打开Terminal(command+空格输入Terminal),准备开始输入命令安装依赖。

当然,我更推荐大家用VSCode编辑器,把本文代码Copy下来,在编辑器下方的终端运行命令安装依赖模块,多舒服的一件事啊:Python 编程的最好搭档—VSCode 详细指南。

在终端输入以下命令安装我们所需要的依赖模块:

  1. pip install scikit-learn 

2.简单的训练集

冬天快到了,深圳这几天已经准备开始入冬了。

从生活入手,外界温度对是否穿外套的影响是具有线性关系的:

现在,考虑这样的一个问题:如果深圳的温度是12度,我们应不应该穿外套?

这个问题很简单,上述简单的训练集中,我们甚至不需要机器学习就能轻易地得到答案:应该。但如果训练集变得稍显复杂一些呢:

你能看出其中x1, x2, x3和y之间的规律吗?

比较难,但是如果你有足够的数据(比如100个),机器学习能够迅速解决这个问题。

为了方便展示机器学习的威力,我们在这里生产100个这样的训练集(公式为: y=x1 + 2*x2 + 3*x3):

  1. from random import randint 
  2. TRAIN_SET_LIMIT = 1000 
  3. TRAIN_SET_COUNT = 100 
  4.  
  5. TRAIN_INPUT = list() 
  6. TRAIN_OUTPUT = list() 
  7. for i in range(TRAIN_SET_COUNT): 
  8.     a = randint(0, TRAIN_SET_LIMIT) 
  9.     b = randint(0, TRAIN_SET_LIMIT) 
  10.     c = randint(0, TRAIN_SET_LIMIT) 
  11.     op = a + (2*b) + (3*c) 
  12.     TRAIN_INPUT.append([a, b, c]) 
  13.     TRAIN_OUTPUT.append(op) 

然后让线性回归模型使用该训练集(Training Set)进行训练(fit),然后再给定三个参数(Test Data),进行预测(predict),让它得到y值(Prediction),如下图所示。

3. 训练和测试

为什么我使用sklearn?因为它真的真的很方便。像这样的训练行为,你只需要3行代码就能搞定:

  1. from sklearn.linear_model import LinearRegression 
  2.  
  3. predictor = LinearRegression(n_jobs=-1) 
  4. predictor.fit(X=TRAIN_INPUTy=TRAIN_OUTPUT

需要注意线性回归模型(LinearRegression)的参数:

n_jobs:默认为1,表示使用CPU的个数。当-1时,代表使用全部CPU

predictor.fit 即训练模型,X是我们在生成训练集时的TRAIN_INPUT,Y即TRAIN_OUTPUT.

训练完就可以立即进行测试了,调用predict函数即可:

  1. X_TEST = [[10, 20, 30]] 
  2. outcome = predictor.predict(X=X_TEST
  3. coefficients = predictor.coef_ 
  4.  
  5. print('Outcome : {}\nCoefficients : {}'.format(outcome, coefficients)) 

这里的 coefficients 是指系数,即x1, x2, x3.

得到的结果如下:

  1. Outcome : [ 140.] 
  2. Coefficients : [ 1. 2. 3.] 

验证一下:10 + 20*2 + 30*3 = 140 完全正确。

如何,机器学习模型,用起来其实真的没你想象中的那么难,大部分人很可能只是卡在了安装 scikit-learn 的路上...

顺便给大家留个小练习,将下列欧式距离,使用线性回归模型进行表示。

解决思路和本文的方案其实是类似的,只不过需要变通一下。

 

责任编辑:赵宁宁 来源: Python实用宝典
相关推荐

2020-12-20 20:31:56

Python机器学习多元线性回归

2020-12-19 10:54:25

机器学习线性回归算法

2014-08-22 10:06:46

机器学习

2020-08-25 18:10:22

Python代码线性回归

2017-05-17 08:24:08

TensorFlow机器学习线性回归

2020-07-01 08:48:01

Python机器学习工具

2014-06-30 10:58:02

机器学习

2024-03-01 07:26:49

RustPolars机器学习

2020-12-23 07:54:56

Python机器学习逻辑回归算法

2021-07-29 13:06:29

Python机器学习编程语言

2022-09-25 23:10:53

Python数据集机器学习

2019-10-11 08:46:45

Python数据Numpy

2023-11-10 09:41:44

Python代码

2021-04-18 22:06:29

机器学习算法数据

2021-05-11 20:46:17

Python代码分类

2017-10-24 14:32:48

2020-12-22 07:09:01

Python机器学习多项式回归

2018-01-10 17:06:36

Python线性回归数据

2017-09-01 13:19:21

机器学习Logostic回归

2017-06-14 17:18:43

机器学习数据研究回归
点赞
收藏

51CTO技术栈公众号