线性回归是机器学习中的一种基本算法,用于预测连续值。以下是一个使用 Python 实现线性回归的示例。

数据准备

首先,我们需要准备一些数据。以下是一个简单的数据集,其中包含自变量 ( x ) 和因变量 ( y )。

x y
1 2
2 4
3 6
4 8
5 10

代码实现

以下是一个使用 Python 和 Jupyter Notebook 实现线性回归的示例。

# 导入必要的库
import numpy as np
import matplotlib.pyplot as plt
from sklearn.linear_model import LinearRegression

# 创建数据集
x = np.array([1, 2, 3, 4, 5]).reshape(-1, 1)
y = np.array([2, 4, 6, 8, 10])

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

# 训练模型
model.fit(x, y)

# 预测
y_pred = model.predict(x)

# 绘制结果
plt.scatter(x, y, color='blue')
plt.plot(x, y_pred, color='red')
plt.xlabel('x')
plt.ylabel('y')
plt.title('线性回归')
plt.show()

结果分析

运行上述代码后,我们可以看到一条红色的线,它代表了线性回归模型的预测结果。从图中可以看出,模型很好地拟合了数据。

线性回归结果图

扩展阅读

如果您想了解更多关于线性回归的信息,可以参考以下链接:

希望这个示例能帮助您更好地理解线性回归。