当前位置：首页 > article >正文

机器学习与数据挖掘_使用梯度下降法训练线性回归模型

article 2024/11/8 8:34:49

实验内容

实验步骤

1. 导入必要的库

2. 加载数据并绘制散点图

3. 设置模型的超参数

4. 实现梯度下降算法

5. 打印训练后的参数和损失值

6. 绘制损失函数随迭代次数的变化图

7. 绘制线性回归拟合曲线

8. 基于训练好的模型进行新样本预测

实验代码

实验结果

实验总结

实验内容

（1）编写代码实现基于梯度下降的单变量线性回归算法，包括梯度的计算与验证；

（2）绘制数据散点图，以及得到的直线；

（3）绘制梯度下降过程中损失的变化图；

（4）基于训练得到的参数，输入新的样本数据，输出预测值。

实验步骤

1. 导入必要的库

使用 `numpy` 进行科学计算，并使用 `matplotlib` 来生成图形。为了保证图形中的中文正常显示，设置 `matplotlib` 的字体为黑体，并解决负号显示问题。

2. 加载数据并绘制散点图

使用 `numpy` 的 `genfromtxt` 函数从文件中加载数据，数据以逗号作为分隔符。分别提取第一列数据为 `x` 值，第二列数据为 `y` 值，展示数据点的分布情况。使用 `scatter` 函数绘制散点图，并使用 `show` 函数显示图形。

3. 设置模型的超参数

初始化线性回归模型的参数：学习率 `alpha` 设置为 `0.0001`。权重 `w` 和偏置 `b` 初始化为 `0`。设置梯度下降的迭代次数为 `1000`。获取数据样本数量 `m`。

4. 实现梯度下降算法

定义一个列表 `MSE` 用来存储每次迭代的均方误差。在每次迭代中，分别计算损失函数和模型参数的梯度：对每一个样本点，计算当前的预测值和真实值的误差，进而计算平方误差并累积。计算梯度，分别对权重 `w` 和偏置 `b` 进行更新。更新后的参数 `w` 和 `b` 基于学习率和当前梯度值来进行调整。

5. 打印训练后的参数和损失值

在训练结束后，打印出模型的最终参数 `w` 和 `b`。使用最后一次迭代的均方误差来表示最终的损失函数值。

6. 绘制损失函数随迭代次数的变化图

使用 `plot` 函数绘制损失函数随迭代次数变化的曲线，`x` 轴为迭代次数，`y` 轴为损失值。图形展示了梯度下降过程中损失函数值的变化趋势，验证模型的收敛情况。

7. 绘制线性回归拟合曲线

再次绘制原始数据的散点图，并基于训练得到的参数计算每个数据点的预测值。使用 `plot` 函数绘制线性回归拟合的曲线，并用红色标出拟合的直线。

8. 基于训练好的模型进行新样本预测

输入新的样本数据 `new_sample`，基于训练得到的参数 `w` 和 `b` 计算新的 `y` 值。打印出新样本数据及其对应的预测值。

实验代码

# 导入必要的库
import numpy as np  # 导入科学计算库
import matplotlib.pyplot as plt  # 导入绘图库
from matplotlib import rcParams  # 导入设置绘图样式的参数

# 设置字体，防止中文乱码
rcParams['font.sans-serif'] = ['SimHei']  # 使用黑体
rcParams['axes.unicode_minus'] = False  # 解决负号显示问题

# 1. 加载数据并画出散点图
points = np.genfromtxt('data1.txt', delimiter=',')  # 从文件中加载数据，数据以逗号分隔
x = points[:, 0]  # 获取第一列数据作为x
y = points[:, 1]  # 获取第二列数据作为y
plt.scatter(x, y)  # 绘制散点图，展示数据的分布情况
plt.show()

# 2. 设置模型的超参数
alpha = 0.0001  # 学习率
w = 0  # 初始化权重w
b = 0  # 初始化偏置b
num_iter = 1000  # 梯度下降的迭代次数
m = len(points)  # 样本数量

# 3. 梯度下降算法
MSE = []  # 用于保存每次迭代的均方误差
for iteration in range(num_iter):
    # 初始化梯度的和
    sum_grad_w = 0  # 用于累加w的梯度
    sum_grad_b = 0  # 用于累加b的梯度
    total_cost = 0  # 每次迭代的总损失初始化为0

    # 遍历所有数据点，计算偏导数并更新梯度
    for i in range(m):
        x_i = points[i, 0]  # 当前数据点的x值
        y_i = points[i, 1]  # 当前数据点的y值

        # 计算当前点的预测值
        pred_y_i = w * x_i + b

        # 计算损失函数（平方误差）
        total_cost += (y_i - pred_y_i) ** 2

        # 计算梯度
        sum_grad_w += (pred_y_i - y_i) * x_i  # 对w的偏导数
        sum_grad_b += (pred_y_i - y_i)  # 对b的偏导数

    # 计算当前迭代的均方误差
    total_cost /= m
    MSE.append(total_cost)  # 保存每次迭代的损失值

    # 计算偏导数的平均值
    grad_w = 2 / m * sum_grad_w
    grad_b = 2 / m * sum_grad_b

    # 更新w和b，基于学习率和梯度
    w -= alpha * grad_w
    b -= alpha * grad_b

# 4. 打印训练后的参数和损失值
print("参数w = ", w)
print("参数b = ", b)
# 使用 MSE[-1] 来表示最后一次迭代的损失函数值
print("最后的损失函数 = ", MSE[-1])

# 5. 绘制损失函数随迭代次数的变化图
plt.plot(MSE)
plt.xlabel('迭代次数')
plt.ylabel('损失值')
plt.title('梯度下降过程中的损失函数变化')
plt.show()

# 6. 画出拟合曲线
plt.scatter(x, y)  # 原始数据的散点图
pred_y = w * x + b  # 基于最终的w和b计算所有数据点的预测值
plt.plot(x, pred_y, color='red')  # 绘制线性回归拟合的直线，颜色为红色
plt.title('线性回归拟合曲线')
plt.show()

# 7. 基于训练得到的参数进行新样本预测
new_sample = np.array([5, 10, 15])  # 新的输入数据
predicted_y = w * new_sample + b  # 计算新样本的预测值
print("输入的新样本数据: ", new_sample)
print("预测的y值: ", predicted_y)