[Matlab科学计算] 共轭梯度法求一般函数的最小值

代码如下: 

function [x,val,k]=frcg
%功能: 用FR共轭梯度法求解无约束问题:  min f(x)
%x0是初始点, fun, gfun分别是目标函数和梯度
%输出:  x, val分别是近似最优点和最优值,  k是迭代次数.
x0 =[2 2]';
maxk=5000;         %最大迭代次数
rho=0.6;sigma=0.4;
k=0;  epsilon=1e-4; 
n=length(x0);
while(k<maxk)
    g=feval('gfun',x0);  %计算梯度
    itern=k-(n+1)*floor(k/(n+1));
    itern=itern+1;
    %计算搜索方向
    if(itern==1)  
        d=-g;  
    else
        beta=(g'*g)/(g0'*g0);
        d=-g+beta*d0;  gd=g'*d;
        if(gd>=0.0)
            d=-g;  
        end
    end
    if(norm(g)<epsilon), break; end   %检验终止条件
    m=0; mk=0;
    while(m<20)   %Armijo搜索
        if(feval('fun',x0+rho^m*d)<feval('fun',x0)+sigma*rho^m*g'*d)
            mk=m; break;
        end
        m=m+1;
    end
    x0=x0+rho^mk*d;
    val=feval('fun',x0);
    g0=g;  d0=d; 
    k=k+1;
end
x=x0;
val=feval('fun',x);

function f=fun(x)
f=(x(1)-1)^2+(x(1)+x(2)+3)^2;

function gf=gfun(x)
gf=[2*(2*x(1)+x(2)+2), 2*(x(1)+x(2)+3)]';

使用说明:

1.根据计算函数修改fun(x)和gfun(x)函数表达式;

2.更改初始点x0的取值。

猜你喜欢

转载自blog.csdn.net/zhwzhaowei/article/details/102589724
今日推荐