逻辑回归的向量化实现样例
From Ufldl
Line 35: | Line 35: | ||
但是,或许可以向量化得更彻底些。如果去除for循环,我们就可以显著地改善代码执行效率。特别的,假定b是一个列向量,A是一个矩阵,我们用以下两种方式来计算A*b: | 但是,或许可以向量化得更彻底些。如果去除for循环,我们就可以显著地改善代码执行效率。特别的,假定b是一个列向量,A是一个矩阵,我们用以下两种方式来计算A*b: | ||
- | % | + | % 矩阵-向量乘法运算的低效代码 |
grad = zeros(n+1,1); | grad = zeros(n+1,1); | ||
for i=1:m, | for i=1:m, | ||
Line 41: | Line 41: | ||
end; | end; | ||
- | % | + | % 矩阵-向量乘法运算的高效代码 |
grad = A*b; | grad = A*b; | ||