梯度下降算法(Gradient descent)

Matlab代码:

% parameter
% 梯度下降法, gradient descent
% heart_scale 数据, 
% Iteration fuction: x(n+1)= x(n)- alpha*grad
% f(x)= 1/2*||y- Ax||^2, 让f(x)最小


% create data
A= rand(20);
y= rand(20, 1);
x= rand(20, 1); % initialize x


% parameter
alpha= 0.001;
iter= 1;
error= 1; 
maxiter= 300;


while error> 1e-1 && iter< maxiter
    
    f(iter)= 1/2*(norm(A*x- y))^2; 
    x_new= x- alpha*(A'*A*x- A'*y);
    
    newf= 1/2*(norm(A*x_new- y))^2; 
    
    fprintf('The %dth iteration, f = %f, \n', iter, f(iter));
    
    error= f(iter)- newf;
    x= x_new;
    iter= iter+ 1; 
end


plot(f, 'r-')
xlabel('Number of iteration')
ylabel('Value of function')
legend('Gradient descent')


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值