matlab - 梯度下降码的向量化

Question

我正在 Matlab 上实现批量梯度下降。我的更新步骤有问题theta。 theta是两个分量（两行）的向量。 X是一个包含m行（训练样本数）和n=2列（特征数）的矩阵。Y 是m行向量。

在更新步骤中，我需要将每个设置theta(i)为

theta(i) = theta(i) - (alpha/m)*sum((X*theta-y).*X(:,i))

这可以通过for循环来完成，但我不知道如何对其进行矢量化（因为这个X(:,i)术语）。

有什么建议吗？

Answer 1

看起来你正在尝试做一个简单的矩阵乘法，这是 MATLAB 最擅长的。

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

Answer 2

除了 Mad Physicist 给出的答案之外，还可以应用以下内容。

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