这个方案对于随机梯度下降的逻辑回归是否正确

数据挖掘 机器学习 分类 逻辑回归 梯度下降
2022-02-28 06:43:28

我正在使用随机梯度下降实现逻辑回归,但它没有按预期工作。我已经尝试了许多时期和不同的学习率但属于正确类别的概率在左右振荡。我有两个表示为的类和两个特征。这个方案正确吗?:α0.50,1

将权重初始化为某个随机数(我选择 1)并循环:w=[w0,w1,w2]

  1. 选择随机训练样本xi=[1,x1,x2]
  2. 计算它属于第1,
    P(y^=1)=11+ewxi
  3. 根据
    w=w+αxi(yiP(y^=1))

这个方案对我来说很有意义,因为如果属于类的概率很低,如果正确的类是 ,那么对权重的更新会很大,但我无法验证它更多。11

1个回答

很难说没有更多细节,但你的更新是不是错了?您需要减去而不是添加渐变。除非 alpha 为负,否则这是错误的。