机器学习100天学习计划 — 第2天 线性回归

一起学！机器学习100天学习计划 (第1天 数据预处理)

今天是机器学习100天学习计划的第2天，我们将实现一个简单的线性回归模型。

线性回归模型就是基于单一特征（X）来预测结果（Y），回归任务的难点在于找到最佳的拟合线，而我们使用机器学习训练模型的目的就是为了找到这条最佳的拟合线。

构建模型的整个流程如下：

第零步：准备

开始之前，你要确保Python和pip已经成功安装在电脑上，如果没有，请访问这篇文章：超详细Python安装指南 进行安装。

如果你用Python的目的是数据分析，可以直接安装Anaconda：Python数据分析与挖掘好帮手—Anaconda，它内置了Python和pip.

此外，推荐大家用VSCode编辑器：Python 编程的最好搭档—VSCode 详细指南

准备输入命令安装依赖，如果你没有VSCode编辑器，Windows环境下打开 Cmd (开始-运行-CMD)，苹果系统环境下请打开 Terminal (command+空格输入Terminal)，如果你用的是VSCode编辑器或Pycharm，可以直接在下方的Terminal中输入命令：

pip install pandas
pip install numpy
pip install matplotlib
pip install scikit-learn

第一步：数据预处理

按照第一天学习的数据预处理知识，这一步我们将执行以下步骤：

1.导入库
2.导入数据集
3.检查缺失数据
4.划分数据集
5.特征归一化（缩放）

代码如下，studentscores.csv 和本文完整源代码，可在Python实用宝典公众号后台回复：机器学习2 下载。

from sklearn.model_selection import train_test_split
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt

dataset = pd.read_csv('studentscores.csv')
X = dataset.iloc[:, : 1].values
Y = dataset.iloc[:, 1].values

X_train, X_test, Y_train, Y_test = train_test_split(
    X, Y, test_size=1/4, random_state=0)

第二步：训练模型

使用 sklearn 的 LinearRegression 能够很轻易地实现一个线性模型的训练。

我们曾经在这篇文章中讲解过：Python 用5行代码学机器学习—线性回归

from sklearn.linear_model import LinearRegression
regressor = LinearRegression()
regressor = regressor.fit(X_train, Y_train)

第三步：预测结果

基于 sklearn 优秀的代码封装能力，预测测试集的结果仅需要一行代码：

Y_pred = regressor.predict(X_test)

第四步：可视化

单纯看数据不是很直观，我们可以使用 matplotlib 将数据可视化。

在真正的实际生活应用中，使用 matplotlib 进行数据可视化这一步往往是在训练模型之前做的，因为我们拿到数据后该做的第一步是探索性数据分析，也叫EDA分析。

不过因为这是一篇教程似的文章，并没有需要进行探索性数据分析的必要，因此我们略过了EDA分析。

训练集可视化：

plt.scatter(X_train, Y_train, color='red')
plt.plot(X_train, regressor.predict(X_train), color='blue')
plt.show()

测试集可视化：

plt.scatter(X_test, Y_test, color='red')
plt.plot(X_test, regressor.predict(X_test), color='blue')
plt.show()

这样，我们便完成了一次简单线性回归模型的训练和测试，还是比较简单的。

如果你想看进一步的应用，可以阅读这篇文章：

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

我们的文章到此就结束啦，如果你喜欢今天的Python 实战教程，请持续关注Python实用宝典。

有任何问题，可以在公众号后台回复：加群，回答相应红字验证信息，进入互助群询问。

原创不易，希望你能在下面点个赞和在看支持我继续创作，谢谢！

点击下方阅读原文可获得更好的阅读体验

Python实用宝典 (pythondict.com)
不只是一个宝典
欢迎关注公众号：Python实用宝典