Vectorization of a gradient descent code

后端 未结 2 398
庸人自扰
庸人自扰 2020-12-25 08:47

I am implementing a batch gradient descent on Matlab. I have a problem with the update step of theta. theta is a vector of two components (two rows

2条回答
  •  生来不讨喜
    2020-12-25 09:00

    In addition to the answer given by Mad Physicist, the following can also be applied.

    theta = theta - (alpha/m) * sum( (X * theta - y).* X )';

提交回复
热议问题