Matlab向量化编程实现机器学习算法---1

本文介绍了如何将机器学习算法,如逻辑回归,的Matlab代码从非向量化转换为向量化,以提高运行速度。通过比较非向量化、部分向量化和完全向量化的代码实现,强调了矩阵运算在消除for循环、利用并行性方面的优势。特别是,完全向量化的梯度计算代码可以显著提升效率。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >


假设x和θ为向量(n+1维实向量空间),需要计算 z=θ(T)x(θ(T)表示θ的转置),那么可以按以下方式实现:
z = 0;
for i=1:(n+1),
    z = z + theta(i) * x(i);
end;

上述代码为非向量化编程,为了更加简洁和提高运行速度,可以采用以下向量化编程方式:

z = theta' * x;    %theta'表示θ的转置

对Matlab代码进行矢量化的工作很大一部分集中在避免使用for循环上,因为这可以使得Matlab更多地利用代码中的并行性,同时其解释器的计算开销更小。

下面给出一些算法的非向量化和向量化代码对照:

1. 逻辑回归梯度计算

% 代码1---非向量化
grad = zeros(n+1,1);
for i=1:m,
   h = sigmoid(theta'*x(:,i));    % sigmoid为S形函数
   temp = y(i) - h;
   for j=1:n+1,
       grad(j) = grad(j) + temp * x(j,i);
   end;
end;

嵌套的for循环语句使这段代码的运行非常缓慢。以下是更典型的实现方式,它对算法进行部分向量化,带来更优的执行效率:

% 代码2---部分向量化
grad = zeros(n+1,1);
for i=1:m,
     grad = grad + (y(i) - sigmoid(theta'*x(:,i)))* x(:,i);
end;

或许可以向量化得更彻底些。如果去除for循环,我们就可以显著地改善代码执行效率。特别的,假定b是一个列向量,A是一个矩阵,我们用以下两种方式来计算A*b:

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值