每天一个小算法(matlab armijo)

下面是 armijo线搜索+最速下降法的小程序,matlab用的很不熟,费了不少劲。

函数:

function g=fun_obj(x)
syms a b
f = 1/2*a^2+b^2-a*b-2*a;
a=x(1);b=x(2);
g=eval(f);

求梯度:

function g=fun_grad(x)
syms a b
f = 1/2*a^2+b^2-a*b-2*a;
gradient = jacobian(f,[a,b]);
a = x(1);b = x(2);
g = eval(gradient);

armijo线搜索:

function mk = armijo( xk, rho, sigma, d )

assert( rho > 0 && rho < 1 );
assert( sigma > 0 && sigma < 0.5 );

mk = 0; max_mk = 100;

while mk <= max_mk
    x = xk + rho^mk * d;
    if fun_obj( x ) <=  fun_obj( xk ) + sigma * rho^mk *fun_grad(xk)*d';
        break;
    end
    mk = mk + 1;
end

return;

主程序:

function result = armijograd(x0)

max_iter = 5000;    % max number of iterations
EPS = 1e-6;         % threshold of gradient norm
            
rho = 0.45; sigma = 0.2;   % Armijo parameters

k = 0; xk = x0; % initialization

while k < max_iter
    
    k = k + 1;
    
    dk =  fun_grad( xk );      % gradient vector
    d = -1 * dk;               % search direction
    
    if norm( dk ) < EPS        %precision
        break;
    end
    
    mk = armijo( xk, rho, sigma, d);   %armijo line search
    
    xk = xk + rho^mk * d;              %update
end
    result = xk;   
return;


最终结果是:[4,2]';程序正确。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值