美文网首页
优化算法matlab实现(三)粒子群算法

优化算法matlab实现(三)粒子群算法

作者: stronghorse | 来源:发表于2022-04-01 14:46 被阅读0次

    注意:此代码实现的是求目标函数最大值,求最小值可将适应度函数乘以-1(框架代码已实现)。
    注意:此代码实现的是求目标函数最大值,求最小值可将适应度函数乘以-1(框架代码已实现)。
    注意:此代码实现的是求目标函数最大值,求最小值可将适应度函数乘以-1(框架代码已实现)。

    1. 代码实现

    实现代码前需要先完成上一篇优化算法matlab实现(二)框架编写中的框架的编写。

    文件名 描述
    :..\optimization algorithm\frame\Unit.m 个体
    :..\optimization algorithm\frame\Algorithm_Impl.m 算法主体

    粒子群算法的每个粒子有两个独特的属性,速度和历史最优位置,在继承后需要单独添加。
    如果不了解粒子群算法可以先看看优化算法笔记(三)粒子群算法(1)
    粒子群算法个体
    文件名:..\optimization algorithm\algorithm_particle_swarm\PSO_Unit.m

    classdef PSO_Unit < Unit
        properties
            % 个体的速度
            velocity
            % 个体的历史最优
            position_best
        end
        
        methods
            function obj = PSO_Unit()
            end
        end
    end
    

    粒子群算法主体
    文件名:..\optimization algorithm\algorithm_particle_swarm\PSO_Base.m

    % 粒子群算法
    classdef PSO_Base < Algorithm_Impl
        %% 算法属性及参数
        properties
            % 算法名称:
            name = 'PSO';
            % 最大速度限制
            velocity_max_list;
            % 自我学习系数
            C1 = 2;
            % 全局学习系数
            C2 = 2;
            % 惯性系数
            W = 1;
        end
        
        %% 外部可调用的方法
        methods
            % 构造函数
            function self = PSO_Base(dim,size,iter_max,range_min_list,range_max_list)
                self@Algorithm_Impl(dim,size,iter_max,range_min_list,range_max_list);
                % 初始化速度上限(为取值范围长度的1/10)
                self.velocity_max_list = (self.range_max_list - self.range_min_list)*0.1;
            end
        end 
        
        %% 继承重写父类的方法
        methods (Access = protected)
            % 初始化种群
            function init(self)
                init@Algorithm_Impl(self)
                % 初始化种群中的每个个体
                for i = 1:self.size 
                    unit = PSO_Unit();
                    % 随机初始化位置:rand(0,1).*(max-min)+min
                    unit.position = unifrnd(self.range_min_list,self.range_max_list);
                    % 初始化时,历史最优就是当前位置
                    unit.position_best = unit.position;
                    % 随机初始化速度:rand(0,1).*(max-(-max))+(-max)
                    unit.velocity = unifrnd(-self.velocity_max_list , self.velocity_max_list);
                    unit.value = self.cal_fitfunction(unit.position);
                    % 将个体加入群体数组
                    self.unit_list = [self.unit_list,unit];
                end
            end
            
            % 每一代的更新
            function update(self,iter)
                update@Algorithm_Impl(self,iter)
                for i = 1:self.size
                    % 更新该个体的速度
                    self.update_velocity(i)
                    % 更新该个体的位置
                    self.update_position(i)
                end
            end
            
            % 更新个体的速度
            function update_velocity(self,id)
                % 获取当前个体实例
                unit = self.unit_list(id);
                % 计算新的速度
                % 公式: v_new = W*v_old+C1*rand(0,1).*(p_best-unit_pos)+C2*rand(0,1).*(g_best-unit_pos)
                velocity_new = self.W*unit.velocity+self.C1*rand(1,self.dim).*(unit.position_best-unit.position)+self.C2*rand(1,self.dim).*(self.position_best-unit.position);
                velocity_new = self.get_out_bound_value(velocity_new,-self.velocity_max_list,self.velocity_max_list);
                % 修改该个体速度
                unit.velocity = velocity_new;
                % 保存修改值
                self.unit_list(id) = unit;
            end
            
            % 更新个体的位置
            function update_position(self,id)
                unit = self.unit_list(id);
                % 计算出新的位置
                position_new = unit.position + unit.velocity;
                % 个体移动到该位置
                unit.position = position_new;
                % 计算该位置的适应度值
                value = self.cal_fitfunction(unit.position);
                if (value > unit.value)
                    % 只记录历史最优值
                    unit.value = value;
                    % 由于历史最优值时更新历史最优位置
                    unit.position_best = unit.position;
                end
                % 保存修改值
                self.unit_list(id) = unit;
            end
           %%
        end
    end
    

    文件名:..\optimization algorithm\algorithm_particle_swarm\PSO_Impl.m
    算法实现,继承于Base,图方便也可不写,直接用PSO_Base,这里为了命名一致。

    %PSO实现
    classdef PSO_Impl < PSO_Base
    
        % 外部可调用的方法
        methods
            function self = PSO_Impl(dim,size,iter_max,range_min_list,range_max_list)
                % 调用父类构造函数设置参数
                 self@PSO_Base(dim,size,iter_max,range_min_list,range_max_list);
            end
        end 
    end
    

    2. 测试

    写一个简单的测试函数试一试代码能否正常运行。
    文件名:..\optimization algorithm\algorithm_particle_swarm\Test_Function.m
    测试函数求(x-10)的平方和的最小值

    function value = Test_Function( x )
        value=sum((x-10).^2);
    end
    

    测试代码
    文件名:..\optimization algorithm\algorithm_particle_swarm\Test.m

    %% 清理之前的数据
    % 清除所有数据
    clear all;
    % 清除窗口输出
    clc;
    
    %% 添加框架路径
    % 将上级目录中的frame文件夹加入路径
    addpath('../frame')
    
    %% 算法实例
    % 算法维度
    dim = 2;
    % 种群数量
    size = 20;
    % 最大迭代次数
    iter_max = 50;
    % 取值范围上界
    range_max_list = ones(1,dim)*100;
    % 取值范围下界
    range_min_list = -ones(1,dim)*100;
    
    % 实例化粒子群类
    base = PSO_Impl(dim,size,iter_max,range_min_list,range_max_list);
    %告诉算法目标函数求的是最小值,不是求最大值
    base.is_cal_max = false;
    % 确定适应度函数
    base.fitfunction = @Test_Function;
    % 运行
    base.run();
    
    % 添加负号是因为目标函数是求最小值
    disp(['最优值',num2str(-base.value_best)])
    disp(['最优解',num2str(base.position_best)])
    
    % 画一个简单的图像
    plot(base.value_best_history);
    

    相关文章

      网友评论

          本文标题:优化算法matlab实现(三)粒子群算法

          本文链接:https://www.haomeiwen.com/subject/assajrtx.html