pytorch梯度下降函数_numpy和pytorch实现梯度下降法

1 梯度下降原理

首先,我们有一个可微分的函数。这个函数就代表着一座山。我们的目标就是找到这个函数的最小值,也就是山底。根据之前的场景假设,最快的下山的方式就是找到当前位置最陡峭的方向,然后沿着此方向向下走,对应到函数中,就是找到给定点的梯度 ,然后朝着梯度相反的方向,就能让函数值下降的最快!

所以,我们重复利用这个方法,反复求取梯度,最后就能到达局部的最小值,这就类似于我们下山的过程。而求取梯度就确定了最陡峭的方向,也就是场景中测量方向的手段。

1.1 梯度

梯度就是分别对每个变量进行微分,然后用逗号分割开,梯度是用<>包括起来,说明梯度其实一个向量。

梯度是微积分中一个很重要的概念,梯度的意义:

在单变量的函数中,梯度其实就是函数的微分,代表着函数在某个给定点的切线的斜率;

在多变量函数中,梯度是一个向量,向量有方向,梯度的方向就指出了函数在给定点的上升最快的方向.

梯度的方向是函数在给定点上升最快的方向,那么梯度的反方向就是函数在给定点下降最快的方向,这正是我们所需要的。所以我们只要沿着梯度的方向一直走,就能走到局部的最低点!

1.2 梯度的数学解释

此公式的意义是:J是关于Θ的一个函数,我们当前所处的位置为Θ0点,要从这个点走到J的最小值点,也就是山底。首先我们先确定前进的方向,也就是梯度的反向,然后走一段距离的步长,也就是α,走完这个段步长,就到达了Θ1这个点!

2 梯度下降算法实现——线性回归模型

2.1 numpy

定义一个代价函数,在此我们选用均方误差代价函数

m是数据集中点的个数

½是一个常量,这样是为了在求梯度的时候,二次方乘下来就和这里的½抵消了,自然就没有多余的常数系数,方便后续的计算,同时对结果不会有影响

y 是数据集中每个点的真实y坐标的值

h 是我们的预测函数,根据每一个输入x,根据Θ 计算得到预测的y值,即

我们有两个变量,为了对这个公式进行矩阵化,我们可以给每一个点x增加一维,这一维的值固定为1,这一维将会乘到Θ0上。这样就方便我们统一矩阵化的计算

我们将代价函数和梯度转化为矩阵向量相乘的形式

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
如果您想使用梯度下降法来最小化损失函数并更新分布概率,您可以使用PyTorch的自动求导功能和优化器来实现。以下是一个示例代码: ```python import torch import torch.nn.functional as F # 初始化概率分布 probabilities = torch.tensor([0.25, 0.25, 0.25, 0.25], requires_grad=True) # 定义损失函数 def loss_function(p): target = torch.tensor([0.1, 0.3, 0.5, 0.1]) return F.mse_loss(p, target) # 定义优化器 optimizer = torch.optim.SGD([probabilities], lr=0.1) # 迭代更新概率分布 num_epochs = 100 for epoch in range(num_epochs): # 计算损失函数 loss = loss_function(probabilities) # 反向传播和优化 optimizer.zero_grad() loss.backward() optimizer.step() # 打印损失值和更新的概率分布 print('Epoch [{}/{}], Loss: {:.4f}'.format(epoch+1, num_epochs, loss.item())) print('Updated Probabilities:', probabilities.detach().numpy()) ``` 在这个例子中,我们首先初始化一个概率分布`probabilities`,并将`requires_grad`设置为True以启用自动求导。 然后,我们定义了一个损失函数`loss_function`,它计算预测的概率分布与目标概率分布之间的均方误差损失。 接下来,我们创建了一个优化器`optimizer`,它使用随机梯度下降方法来更新概率分布。 在训练循环中,我们计算损失函数进行反向传播和优化。然后,我们打印每一轮的损失值和更新的概率分布。 请注意,这里的目标概率分布`target`是根据实际情况定义的。 希望对您有所帮助!如有任何疑问,请随时提问。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值