2 反向传播和线性回归

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
岭回归是一种用于处理具有共线性的数据的线性回归方法。它通过在损失函数中加入正则化项来避免过拟合。具体来说,损失函数可以表示为: $J(w) = \frac{1}{2n} \sum_{i=1}^{n}(y_i - \hat{y_i})^2 + \frac{\lambda}{2}||w||^2$ 其中,$w$ 是模型参数,$y_i$ 是第 $i$ 个样本的真实标签,$\hat{y_i}$ 是模型对第 $i$ 个样本的预测值,$n$ 是样本数量,$\lambda$ 是正则化系数,$||w||^2$ 是参数 $w$ 的 L2 范数的平方。 为了实现岭回归,我们可以使用 PyTorch 提供的自动求导功能来计算梯度并更新参数。下面是一个简单的实现: ```python import torch # 构造数据 x = torch.randn(100, 10) y = torch.randn(100, 1) # 定义模型 model = torch.nn.Linear(10, 1) # 定义损失函数和正则化系数 criterion = torch.nn.MSELoss() lambda_ = 0.1 # 定义优化器 optimizer = torch.optim.SGD(model.parameters(), lr=0.1) # 训练模型 for epoch in range(100): # 前向传播 y_pred = model(x) # 计算损失函数和正则化项 loss = criterion(y_pred, y) l2_reg = lambda_ * torch.norm(model.weight, p=2) loss += l2_reg # 反向传播 optimizer.zero_grad() loss.backward() optimizer.step() # 打印损失函数 print('Epoch [{}/{}], Loss: {:.4f}'.format(epoch+1, 100, loss.item())) ``` 在每个 epoch 中,我们首先进行前向传播计算预测值 $y_{pred}$,然后计算损失函数 $loss$ 和正则化项 $l2\_reg$。接着进行反向传播计算梯度并更新模型参数,最后打印出当前的损失函数。 需要注意的是,在计算正则化项时,我们使用了 PyTorch 提供的 `torch.norm` 函数来计算参数 $w$ 的 L2 范数。此外,我们使用了随机梯度下降(SGD)优化器来更新参数,但也可以选择其他优化器,比如 Adam 等。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值