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

PyTorch线性回归实战：从数据生成到模型评估

news 2026/8/19 9:18:23

1. 项目概述：PyTorch线性回归实战全流程

线性回归作为机器学习领域的"Hello World"，是每个从业者必须掌握的基础模型。不同于教科书式的理论讲解，这次我们直接用PyTorch实现从数据生成到模型评估的完整流程。选择PyTorch而非其他框架的原因很简单——它的动态计算图机制让调试过程直观可见，特别适合教学演示。我在工业界参与过多个预测类项目，发现很多复杂问题经过特征工程后，本质上仍可转化为线性回归问题。

本次实战将重点解决三个核心问题：如何生成符合真实场景的模拟数据？如何设计合理的训练循环？以及如何解读评估指标？这些技能在房价预测、销量预估等场景中都有直接应用价值。即使你刚接触机器学习，只要熟悉Python基础语法就能跟上节奏。

2. 环境配置与数据生成

2.1 PyTorch环境搭建

推荐使用conda创建隔离环境，避免包冲突。对于CUDA版本选择，当前主流显卡建议搭配PyTorch 2.0+和CUDA 11.8：

conda create -n torch_reg python=3.9 conda activate torch_reg conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

注意：如果使用AMD显卡，需要安装ROCm版本的PyTorch。可通过torch.cuda.is_available()验证GPU是否可用。

2.2 数据生成策略

真实场景的数据往往包含噪声和异常值。我们生成1000个样本，包含以下特征：

基础线性关系：y = 2X + 1
添加高斯噪声：标准差0.5
5%的异常值：偏离均值3个标准差

import torch import numpy as np def generate_data(n_samples=1000): X = torch.linspace(0, 10, n_samples).unsqueeze(1) y = 2 * X + 1 # 添加噪声 noise = torch.randn(X.shape) * 0.5 y += noise # 添加异常值 outlier_mask = torch.rand(len(X)) < 0.05 y[outlier_mask] += torch.randn(outlier_mask.sum()) * 3 return X, y X, y = generate_data()

可视化生成的数据（使用matplotlib）：

plt.scatter(X.numpy(), y.numpy(), s=5, label='data') plt.plot(X.numpy(), 2*X.numpy()+1, c='r', label='true') plt.legend()

3. 模型构建与训练

3.1 线性回归实现

PyTorch提供两种实现方式：

继承nn.Module类（推荐）
直接使用nn.Linear

我们采用第一种方式，便于后续扩展：

class LinearRegression(nn.Module): def __init__(self): super().__init__() self.linear = nn.Linear(1, 1) # 输入输出维度均为1 def forward(self, x): return self.linear(x)

3.2 训练超参数配置

关键参数选择依据：

学习率0.01：经过网格搜索验证的效果
批次大小32：兼顾内存和梯度稳定性
迭代次数100：观察损失曲线已收敛

model = LinearRegression() criterion = nn.MSELoss() # 均方误差损失 optimizer = torch.optim.SGD(model.parameters(), lr=0.01) # 数据划分 X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2)

3.3 训练循环实现

加入早停机制防止过拟合：

best_loss = float('inf') patience = 5 counter = 0 for epoch in range(100): # 训练模式 model.train() optimizer.zero_grad() outputs = model(X_train) loss = criterion(outputs, y_train) loss.backward() optimizer.step() # 验证模式 model.eval() with torch.no_grad(): val_loss = criterion(model(X_test), y_test) # 早停判断 if val_loss < best_loss: best_loss = val_loss counter = 0 else: counter += 1 if counter >= patience: print(f'Early stopping at epoch {epoch}') break

4. 模型评估与可视化

4.1 评估指标计算

除了基础的MSE，建议计算：

R²分数：解释方差比例
MAE：对异常值更鲁棒

from sklearn.metrics import r2_score def evaluate(model, X, y): with torch.no_grad(): preds = model(X) mse = criterion(preds, y) mae = torch.abs(preds - y).mean() r2 = r2_score(y.numpy(), preds.numpy()) return {'MSE': mse.item(), 'MAE': mae.item(), 'R2': r2}

4.2 结果可视化技巧

动态绘制训练过程（需要IPython环境）：

from IPython import display def live_plot(): plt.clf() plt.scatter(X_test, y_test, c='b', s=5, label='data') plt.plot(X_test, model(X_test).detach(), c='r', label='pred') plt.legend() display.clear_output(wait=True) display.display(plt.gcf())

4.3 权重分析

检查学习到的参数是否符合预期：

weight = model.linear.weight.item() bias = model.linear.bias.item() print(f'Learned weights: w={weight:.2f}, b={bias:.2f}') print(f'True weights: w=2.00, b=1.00')

5. 工业级优化技巧

5.1 数据标准化

虽然简单线性回归不需要，但养成标准化习惯对复杂模型很重要：

from sklearn.preprocessing import StandardScaler scaler = StandardScaler() X_scaled = scaler.fit_transform(X)

5.2 学习率调度

动态调整学习率提升收敛速度：

scheduler = torch.optim.lr_scheduler.ReduceLROnPlateau( optimizer, mode='min', factor=0.1, patience=3)

5.3 梯度裁剪

防止梯度爆炸：

torch.nn.utils.clip_grad_norm_(model.parameters(), max_norm=1.0)

6. 常见问题排查

6.1 损失不下降的可能原因

现象	排查方向	解决方案
损失震荡	学习率过大	逐步降低学习率
损失不变	梯度消失	检查初始化权重
指标异常	数据泄漏	验证数据划分

6.2 GPU相关错误处理

# 设备自动选择 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') model = model.to(device) X, y = X.to(device), y.to(device)

6.3 模型保存与加载

# 保存 torch.save({ 'model_state': model.state_dict(), 'optimizer_state': optimizer.state_dict() }, 'regression.pth') # 加载 checkpoint = torch.load('regression.pth') model.load_state_dict(checkpoint['model_state'])