首页 > 文章列表 > Python中的简单线性回归实例

Python中的简单线性回归实例

Python 实例 线性回归
245 2023-06-13

简介

线性回归是统计学中用于建立两个变量之间线性关系的方法之一,它用于预测一个变量在另一个变量上的取值。Python中有许多机器学习库可以用于实现线性回归,本文将演示如何使用Python进行简单线性回归实例。

实现

下面的代码演示如何使用Python的Scikit-Learn库进行简单线性回归:

import numpy as np
from sklearn.linear_model import LinearRegression

# 定义输入和输出向量
x = np.array([5, 15, 25, 35, 45, 55]).reshape((-1, 1))
y = np.array([5, 20, 14, 32, 22, 38])

# 创建一个线性回归模型
model = LinearRegression()

# 将数据拟合到模型中
model.fit(x, y)

# 预测值
y_pred = model.predict(x)

# 打印模型参数
print('斜率:', model.coef_)
print('截距:', model.intercept_)

在上面的代码中,首先导入必要的库(numpy和Scikit-Learn)。然后定义输入和输出向量xy,这里x表示自变量,y表示因变量。 然后创建一个线性回归模型。最后,将数据拟合到模型中,并进行预测,最后打印出模型参数。

结果:

斜率: [0.54]
截距: 5.633333333333329

此外,我们还可以使用Matplotlib库将输入和输出数据绘制在图表中,代码如下所示:

import matplotlib.pyplot as plt

# 绘制数据点
plt.scatter(x.reshape(1,-1), y, color='black')

# 绘制线性回归线
plt.plot(x.reshape(1,-1), y_pred, color='blue', linewidth=2)

# 显示图表
plt.show()

将数据点和回归线绘制在图表中,即可进行可视化。

总结

本文介绍了如何使用Python中的Scikit-Learn库进行简单线性回归的实现。希望通过本文的介绍和代码示例,读者能够更加深入地理解简单线性回归的操作步骤和流程。对于掌握线性回归技术的读者,可以继续学习多项式回归、岭回归等高级模型的使用方法。