机器学习第1天：线性回归（代码篇）

一、具体实现步骤

1. 导入Iris鸢尾花数据集

Iris鸢尾花数据集共有150条记录，分别是：

50条山鸢尾 (Iris-setosa)
50条变色鸢尾(Iris-versicolor)
50条维吉尼亚鸢尾(Iris-virginica)

代码语言：javascript复制

url = "https://archive.ics.uci.edu/ml/machine-learning-databases/iris/iris.data"  
names = ['花萼-length', '花萼-width', '花瓣-length', '花瓣-width', 'class']  
dataset = pd.read_csv(url, names=names)

2. 提取花瓣数据

下面我们提取数据集中花瓣宽度与花瓣长度数据，将花瓣数据分为训练数据与测试数据，训练数据用于训练线性回归模型，测试数据用于检测我们的模型的准确率。

最终我们要达到的效果是：输入花瓣宽度，通过模型预测花瓣宽度。

代码语言：javascript复制

X = dataset["花瓣-length"]
Y = dataset["花瓣-width"]
X = X.reshape(len(X),1)
Y = Y.reshape(len(Y),1)

3. 拆分数据

将数据集拆分数据集成训练集、测试集

代码语言：javascript复制

from sklearn.model_selection import train_test_split
X_train, X_test, Y_train, Y_test = train_test_split(X, Y, test_size=0.2, random_state=0)

4. 训练模型

这里我们需要将我们的训练数据喂给模型进行训练。

代码语言：javascript复制

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

二、可视化结果展示

1. 训练集

将训练集中每一朵花的花瓣数据与线性回归模型预测的结果放到同一张统计图中。

代码语言：javascript复制

import matplotlib.pyplot as plt

plt.scatter(X_train, Y_train, color='red')
plt.plot(X_train, regressor.predict(X_train), color='green')
plt.xlabel("Iris-length") 
plt.ylabel("Iris-width") 
plt.title("This is train dataset-kzb")
plt.show()

红色的点是训练数据集中的花瓣数据，我们不难看出花瓣长度与宽度是一个线性关系，绿色的线是我们模型拟合的结果。

2. 测试集

将测试集中每一朵花的花瓣数据与线性回归模型预测的结果放到同一张统计图中。

代码语言：javascript复制

plt.scatter(X_test, Y_test, color='blue')
plt.plot(X_train, regressor.predict(X_train), color='green')
plt.xlabel("Iris-length") 
plt.ylabel("Iris-width") 
plt.title("This is test dataset-kzb")
plt.show()

绿色的点是测试数据集中的花瓣数据，我们可以看出这部分数据也是符合线性关系的，随着集的增大，线性关系会更加明显。

三、相关知识点讲解

1. train_test_split()函数

train_test_split()：将数据集划分为测试集与训练集。

X：所要划分的整体数据的特征集；
Y：所要划分的整体数据的结果；
test_size：测试集数据量在整体数据量中的占比（可以理解为X_test与X的比值）；
random_state：①若不填或者填0，每次生成的数据都是随机，可能不一样。②若为整数，每次生成的数据都相同；

代码语言：javascript复制

from sklearn.model_selection import train_test_split
X_train, X_test, Y_train, Y_test = train_test_split(X, Y, test_size=0.2, random_state=0)

2. LinearRegression()函数

sklearn.linear_model包实现了广义线性模型，包括线性回归、Ridge回归、Bayesian回归等。LinearRegression是其中较为简单的线性回归模型。

解释一下什么是回归：回归最简单的定义是，给出一个点集D，用一个函数去拟合这个点集，并且使得点集与拟合函数间的误差最小，如果这个函数曲线是一条直线，那就被称为线性回归，如果曲线是一条二次曲线，就被称为二次回归。

3. 散点图与折线统计图的绘制

plt.scatter()：绘画出数据的散点图 plt.plot()：绘画出依据模型（LinearRegression的线性回归模型）生成的直线

更详细的介绍可以参考：【Matplotlib可视化系列教程】

腾讯云测试服务线性回归 javascript

0 人点赞