鲸鱼优化算法代码

赧然的贼158

于 2024-07-17 08:43:24 发布

阅读量198

点赞数 2

文章标签：算法

本文链接：https://blog.csdn.net/weixin_65744632/article/details/140483649

版权

主函数 main

%_________________________________________________________________________%
%  Whale Optimization Algorithm (WOA) source codes demo 1.0               %
%                                                                         %
%  Developed in MATLAB R2011b(7.13)                                       %
%                                                                         %
%  Author and programmer: Seyedali Mirjalili                              %
%                                                                         %
%         e-Mail: ali.mirjalili@gmail.com                                 %
%                 seyedali.mirjalili@griffithuni.edu.au                   %
%                                                                         %
%       Homepage: http://www.alimirjalili.com                             %
%                                                                         %
%   Main paper: S. Mirjalili, A. Lewis                                    %
%               The Whale Optimization Algorithm,                         %
%               Advances in Engineering Software , in press,              %
%               DOI: http://dx.doi.org/10.1016/j.advengsoft.2016.01.008   %
%                                                                         %
%_________________________________________________________________________%

% 你可以简单地在一个单独的文件中定义你的目标函数，并将其句柄加载到fobj中
% 初始参数包括：
%__________________________________________
% fobj = @YourCostFunction
% dim = 变量的数量
% Max_iteration = 最大迭代次数
% SearchAgents_no = 搜索代理的数量
% lb=[lb1,lb2,...,lbn] 每个变量的下边界
% ub=[ub1,ub2,...,ubn] 每个变量的上边界
% 如果所有变量的上下边界相同，可以将lb和ub定义为两个单独的数字

% 运行WOA: [Best_score,Best_pos,WOA_cg_curve]=WOA(SearchAgents_no,Max_iteration,lb,ub,dim,fobj)

clear all % 清除所有变量
clc % 清空命令行窗口

SearchAgents_no=30; % 搜索代理的数量

Function_name='F1'; % 测试函数的名称，可以从F1到F23选择（参考论文中的表1、表2、表3）

Max_iteration=500; % 最大迭代次数

% 加载所选基准函数的详细信息
[lb,ub,dim,fobj]=Get_Functions_details(Function_name);

% 运行鲸鱼优化算法
[Best_score,Best_pos,WOA_cg_curve]=WOA(SearchAgents_no,Max_iteration,lb,ub,dim,fobj);

% 创建一个图形窗口并设置其位置和大小
figure('Position',[269   240   660   290])
% 绘制搜索空间
subplot(1,2,1); % 创建一个1行2列的子图，并选择第一个子图
func_plot(Function_name); % 绘制基准函数的图形
title('Parameter space') % 设置图形标题
xlabel('x_1'); % 设置x轴标签
ylabel('x_2'); % 设置y轴标签
zlabel([Function_name,'( x_1 , x_2 )']) % 设置z轴标签

% 绘制目标空间
subplot(1,2,2); % 选择第二个子图
semilogy(WOA_cg_curve,'Color','r') % 绘制收敛曲线，使用红色
title('Objective space') % 设置图形标题
xlabel('Iteration'); % 设置x轴标签
ylabel('Best score obtained so far'); % 设置y轴标签

axis tight % 紧凑轴
grid on % 打开网格
box on % 打开边框
legend('WOA') % 添加图例

% 显示优化结果
display(['The best solution obtained by WOA is : ', num2str(Best_pos)]);
display(['The best optimal value of the objective funciton found by WOA is : ', num2str(Best_score)]);

WOA

%_________________________________________________________________________%
%  Whale Optimization Algorithm (WOA) source codes demo 1.0               %
%                                                                         %
%  Developed in MATLAB R2011b(7.13)                                       %
%                                                                         %
%  Author and programmer: Seyedali Mirjalili                              %
%                                                                         %
%         e-Mail: ali.mirjalili@gmail.com                                 %
%                 seyedali.mirjalili@griffithuni.edu.au                   %
%                                                                         %
%       Homepage: http://www.alimirjalili.com                             %
%                                                                         %
%   Main paper: S. Mirjalili, A. Lewis                                    %
%               The Whale Optimization Algorithm,                         %
%               Advances in Engineering Software , in press,              %
%               DOI: http://dx.doi.org/10.1016/j.advengsoft.2016.01.008   %
%                                                                         %
%_________________________________________________________________________%

% Whale Optimization Algorithm 主函数
function [Leader_score, Leader_pos, Convergence_curve] = WOA(SearchAgents_no, Max_iter, lb, ub, dim, fobj)

% 初始化领导者的位置向量和得分
Leader_pos = zeros(1, dim); % 领导者的位置向量
Leader_score = inf; % 将领导者得分初始化为无穷大，对于最大化问题则设为-inf

% 初始化搜索代理的位置
Positions = initialization(SearchAgents_no, dim, ub, lb); % 调用初始化函数生成初始位置

Convergence_curve = zeros(1, Max_iter); % 初始化收敛曲线向量

t = 0; % 迭代计数器

% 主循环
while t < Max_iter
    for i = 1:size(Positions, 1) % 遍历每个搜索代理
        
        % 返回超出搜索空间边界的搜索代理
        Flag4ub = Positions(i, :) > ub; % 检查是否超过上边界
        Flag4lb = Positions(i, :) < lb; % 检查是否超过下边界
        Positions(i, :) = (Positions(i, :) .* ~(Flag4ub + Flag4lb)) + ub .* Flag4ub + lb .* Flag4lb; % 超出边界的值设置为边界值
        
        % 计算每个搜索代理的目标函数值
        fitness = fobj(Positions(i, :)); % 计算适应度
        
        % 更新领导者
        if fitness < Leader_score % 对于最大化问题将此处改为 >
            Leader_score = fitness; % 更新领导者得分
            Leader_pos = Positions(i, :); % 更新领导者位置
        end
    end
    
    a = 2 - t * ((2) / Max_iter); % 参数 a 从 2 线性减少到 0
    
    a2 = -1 + t * ((-1) / Max_iter); % 参数 a2 从 -1 线性减少到 -2
    
    % 更新搜索代理的位置
    for i = 1:size(Positions, 1)
        r1 = rand(); % r1 是 [0, 1] 之间的随机数
        r2 = rand(); % r2 是 [0, 1] 之间的随机数
        
        A = 2 * a * r1 - a; % 计算 A
        C = 2 * r2; % 计算 C
        
        b = 1; % 常量 b
        l = (a2 - 1) * rand + 1; % 参数 l
        
        p = rand(); % 随机数 p
        
        for j = 1:size(Positions, 2) % 对每个搜索代理的每个维度进行操作
            
            if p < 0.5 % 如果 p 小于 0.5，使用包围机制
                if abs(A) >= 1
                    rand_leader_index = floor(SearchAgents_no * rand() + 1); % 随机选择一个搜索代理
                    X_rand = Positions(rand_leader_index, :);
                    D_X_rand = abs(C * X_rand(j) - Positions(i, j)); % 计算距离
                    Positions(i, j) = X_rand(j) - A * D_X_rand; % 更新位置
                    
                elseif abs(A) < 1
                    D_Leader = abs(C * Leader_pos(j) - Positions(i, j)); % 计算领导者距离
                    Positions(i, j) = Leader_pos(j) - A * D_Leader; % 更新位置
                end
                
            elseif p >= 0.5 % 如果 p 大于等于 0.5，使用螺旋更新位置
                distance2Leader = abs(Leader_pos(j) - Positions(i, j)); % 计算到领导者的距离
                Positions(i, j) = distance2Leader * exp(b * l) * cos(l * 2 * pi) + Leader_pos(j); % 更新位置
            end
            
        end
    end
    t = t + 1; % 更新迭代次数
    Convergence_curve(t) = Leader_score; % 记录领导者得分
    [t Leader_score] % 显示当前迭代次数和领导者得分
end

Get_Functions_details

%_________________________________________________________________________%
%  Whale Optimization Algorithm (WOA) source codes demo 1.0               %
%                                                                         %
%  Developed in MATLAB R2011b(7.13)                                       %
%                                                                         %
%  Author and programmer: Seyedali Mirjalili                              %
%                                                                         %
%         e-Mail: ali.mirjalili@gmail.com                                 %
%                 seyedali.mirjalili@griffithuni.edu.au                   %
%                                                                         %
%       Homepage: http://www.alimirjalili.com                             %
%                                                                         %
%   Main paper: S. Mirjalili, A. Lewis                                    %
%               The Whale Optimization Algorithm,                         %
%               Advances in Engineering Software , in press,              %
%               DOI: http://dx.doi.org/10.1016/j.advengsoft.2016.01.008   %
%                                                                         %
%_________________________________________________________________________%

% 此函数包含了论文表1、表2和表3中的基准函数的完整信息和实现

% lb 是下边界：lb=[lb_1,lb_2,...,lb_d]
% ub 是上边界：ub=[ub_1,ub_2,...,ub_d]
% dim 是变量的数量（问题的维度）

function [lb, ub, dim, fobj] = Get_Functions_details(F)

% 根据输入参数F，选择相应的基准函数，并设置其下边界、上边界和维度
switch F
    case 'F1'
        fobj = @F1;
        lb = -100;
        ub = 100;
        dim = 30;
        
    case 'F2'
        fobj = @F2;
        lb = -10;
        ub = 10;
        dim = 30;
        
    case 'F3'
        fobj = @F3;
        lb = -100;
        ub = 100;
        dim = 30;
        
    case 'F4'
        fobj = @F4;
        lb = -100;
        ub = 100;
        dim = 30;
        
    case 'F5'
        fobj = @F5;
        lb = -30;
        ub = 30;
        dim = 30;
        
    case 'F6'
        fobj = @F6;
        lb = -100;
        ub = 100;
        dim = 30;
        
    case 'F7'
        fobj = @F7;
        lb = -1.28;
        ub = 1.28;
        dim = 30;
        
    case 'F8'
        fobj = @F8;
        lb = -500;
        ub = 500;
        dim = 30;
        
    case 'F9'
        fobj = @F9;
        lb = -5.12;
        ub = 5.12;
        dim = 30;
        
    case 'F10'
        fobj = @F10;
        lb = -32;
        ub = 32;
        dim = 30;
        
    case 'F11'
        fobj = @F11;
        lb = -600;
        ub = 600;
        dim = 30;
        
    case 'F12'
        fobj = @F12;
        lb = -50;
        ub = 50;
        dim = 30;
        
    case 'F13'
        fobj = @F13;
        lb = -50;
        ub = 50;
        dim = 30;
        
    case 'F14'
        fobj = @F14;
        lb = -65.536;
        ub = 65.536;
        dim = 2;
        
    case 'F15'
        fobj = @F15;
        lb = -5;
        ub = 5;
        dim = 4;
        
    case 'F16'
        fobj = @F16;
        lb = -5;
        ub = 5;
        dim = 2;
        
    case 'F17'
        fobj = @F17;
        lb = [-5, 0];
        ub = [10, 15];
        dim = 2;
        
    case 'F18'
        fobj = @F18;
        lb = -2;
        ub = 2;
        dim = 2;
        
    case 'F19'
        fobj = @F19;
        lb = 0;
        ub = 1;
        dim = 3;
        
    case 'F20'
        fobj = @F20;
        lb = 0;
        ub = 1;
        dim = 6;     
        
    case 'F21'
        fobj = @F21;
        lb = 0;
        ub = 10;
        dim = 4;    
        
    case 'F22'
        fobj = @F22;
        lb = 0;
        ub = 10;
        dim = 4;    
        
    case 'F23'
        fobj = @F23;
        lb = 0;
        ub = 10;
        dim = 4;            
end

end

% 基准函数 F1 到 F23 的具体实现
function o = F1(x)
    o = sum(x.^2);
end

function o = F2(x)
    o = sum(abs(x)) + prod(abs(x));
end

function o = F3(x)
    dim = size(x, 2);
    o = 0;
    for i = 1:dim
        o = o + sum(x(1:i))^2;
    end
end

function o = F4(x)
    o = max(abs(x));
end

function o = F5(x)
    dim = size(x, 2);
    o = sum(100 * (x(2:dim) - (x(1:dim-1).^2)).^2 + (x(1:dim-1) - 1).^2);
end

function o = F6(x)
    o = sum(abs((x + 0.5)).^2);
end

function o = F7(x)
    dim = size(x, 2);
    o = sum([1:dim] .* (x.^4)) + rand;
end

function o = F8(x)
    o = sum(-x .* sin(sqrt(abs(x))));
end

function o = F9(x)
    dim = size(x, 2);
    o = sum(x.^2 - 10 * cos(2 * pi .* x)) + 10 * dim;
end

function o = F10(x)
    dim = size(x, 2);
    o = -20 * exp(-0.2 * sqrt(sum(x.^2) / dim)) - exp(sum(cos(2 * pi .* x)) / dim) + 20 + exp(1);
end

function o = F11(x)
    dim = size(x, 2);
    o = sum(x.^2) / 4000 - prod(cos(x ./ sqrt([1:dim]))) + 1;
end

function o = F12(x)
    dim = size(x, 2);
    o = (pi / dim) * (10 * ((sin(pi * (1 + (x(1) + 1) / 4)))^2) + sum((((x(1:dim-1) + 1) / 4).^2) .* (1 + 10 * ((sin(pi * (1 + (x(2:dim) + 1) / 4)))).^2)) + ((x(dim) + 1) / 4)^2) + sum(Ufun(x, 10, 100, 4));
end

function o = F13(x)
    dim = size(x, 2);
    o = 0.1 * ((sin(3 * pi * x(1)))^2 + sum((x(1:dim-1) - 1).^2 .* (1 + (sin(3 * pi * x(2:dim))).^2)) + ((x(dim) - 1)^2) * (1 + (sin(2 * pi * x(dim)))^2)) + sum(Ufun(x, 5, 100, 4));
end

function o = F14(x)
    aS = [-32 -16 0 16 32 -32 -16 0 16 32 -32 -16 0 16 32 -32 -16 0 16 32 -32 -16 0 16 32; ...
          -32 -32 -32 -32 -32 -16 -16 -16 -16 -16 0 0 0 0 0 16 16 16 16 16 32 32 32 32 32];

    for j = 1:25
        bS(j) = sum((x' - aS(:, j)).^6);
    end
    o = (1 / 500 + sum(1 ./ ([1:25] + bS))).^(-1);
end

function o = F15(x)
    aK = [.1957 .1947 .1735 .16 .0844 .0627 .0456 .0342 .0323 .0235 .0246];
    bK = [.25 .5 1 2 4 6 8 10 12 14 16]; bK = 1 ./ bK;
    o = sum((aK - ((x(1) * (bK.^2 + x(2) * bK)) ./ (bK.^2 + x(3) * bK + x(4)))).^2);
end

function o = F16(x)
    o = 4 * (x(1)^2) - 2.1 * (x(1)^4) + (x(1)^6) / 3 + x(1) * x(2) - 4 * (x(2)^2) + 4 * (x(2)^4);
end

function o = F17(x)
    o = (x(2) - (x(1)^2) * 5.1 / (4 * (pi^2)) + 5 / pi * x(1) - 6)^2 + 10 * (1 - 1 / (8 * pi)) * cos(x(1)) + 10;
end

function o = F18(x)
    o = (1 + (x(1) + x(2) + 1)^2 * (19 - 14 * x(1) + 3 * (x(1)^2) - 14 * x(2) + 6 * x(1) * x(2) + 3 * x(2)^2)) * ...
        (30 + (2 * x(1) - 3 * x(2))^2 * (18 - 32 * x(1) + 12 * (x(1)^2) + 48 * x(2) - 36 * x(1) * x(2) + 27 * (x(2)^2)));
end

function o = F19(x)
    aH = [3 10 30; .1 10 35; 3 10 30; .1 10 35]; cH = [1 1.2 3 3.2];
    pH = [.3689 .117 .2673; .4699 .4387 .747; .1091 .8732 .5547; .03815 .5743 .8828];
    o = 0;
    for i = 1:4
        o = o - cH(i) * exp(-(sum(aH(i, :) .* ((x - pH(i, :)).^2))));
    end
end

function o = F20(x)
    aH = [10 3 17 3.5 1.7 8; .05 10 17 .1 8 14; 3 3.5 1.7 10 17 8; 17 8 .05 10 .1 14];
    cH = [1 1.2 3 3.2];
    pH = [.1312 .1696 .5569 .0124 .8283 .5886; .2329 .4135 .8307 .3736 .1004 .9991; ...
          .2348 .1415 .3522 .2883 .3047 .6650; .4047 .8828 .8732 .5743 .1091 .0381];
    o = 0;
    for i = 1:4
        o = o - cH(i) * exp(-(sum(aH(i, :) .* ((x - pH(i, :)).^2))));
    end
end

function o = F21(x)
    aSH = [4 4 4 4; 1 1 1 1; 8 8 8 8; 6 6 6 6; 3 7 3 7; 2 9 2 9; 5 5 3 3; 8 1 8 1; 6 2 6 2; 7 3.6 7 3.6];
    cSH = [.1 .2 .2 .4 .4 .6 .3 .7 .5 .5];

    o = 0;
    for i = 1:5
        o = o - ((x - aSH(i, :)) * (x - aSH(i, :))' + cSH(i))^(-1);
    end
end

function o = F22(x)
    aSH = [4 4 4 4; 1 1 1 1; 8 8 8 8; 6 6 6 6; 3 7 3 7; 2 9 2 9; 5 5 3 3; 8 1 8 1; 6 2 6 2; 7 3.6 7 3.6];
    cSH = [.1 .2 .2 .4 .4 .6 .3 .7 .5 .5];

    o = 0;
    for i = 1:7
        o = o - ((x - aSH(i, :)) * (x - aSH(i, :))' + cSH(i))^(-1);
    end
end

function o = F23(x)
    aSH = [4 4 4 4; 1 1 1 1; 8 8 8 8; 6 6 6 6; 3 7 3 7; 2 9 2 9; 5 5 3 3; 8 1 8 1; 6 2 6 2; 7 3.6 7 3.6];
    cSH = [.1 .2 .2 .4 .4 .6 .3 .7 .5 .5];

    o = 0;
    for i = 1:10
        o = o - ((x - aSH(i, :)) * (x - aSH(i, :))' + cSH(i))^(-1);
    end
end

function o = Ufun(x, a, k, m)
    o = k .* ((x - a).^m) .* (x > a) + k .* ((-x - a).^m) .* (x < -a);
end

func_plot

%_________________________________________________________________________%
%  Whale Optimization Algorithm (WOA) source codes demo 1.0               %
%                                                                         %
%  Developed in MATLAB R2011b(7.13)                                       %
%                                                                         %
%  Author and programmer: Seyedali Mirjalili                              %
%                                                                         %
%         e-Mail: ali.mirjalili@gmail.com                                 %
%                 seyedali.mirjalili@griffithuni.edu.au                   %
%                                                                         %
%       Homepage: http://www.alimirjalili.com                             %
%                                                                         %
%   Main paper: S. Mirjalili, A. Lewis                                    %
%               The Whale Optimization Algorithm,                         %
%               Advances in Engineering Software , in press,              %
%               DOI: http://dx.doi.org/10.1016/j.advengsoft.2016.01.008   %
%                                                                         %
%_________________________________________________________________________%

% 该函数绘制基准函数的图形
function func_plot(func_name)

    % 获取基准函数的详细信息，包括下边界、上边界、维度和函数句柄
    [lb, ub, dim, fobj] = Get_Functions_details(func_name);

    % 根据函数名称设置绘图范围
    switch func_name 
        case 'F1' 
            x = -100:2:100; y = x; % 设置x和y的范围为[-100, 100]
        case 'F2' 
            x = -100:2:100; y = x; % 设置x和y的范围为[-100, 100]
        case 'F3' 
            x = -100:2:100; y = x; % 设置x和y的范围为[-100, 100]
        case 'F4' 
            x = -100:2:100; y = x; % 设置x和y的范围为[-100, 100]
        case 'F5' 
            x = -200:2:200; y = x; % 设置x和y的范围为[-200, 200]
        case 'F6' 
            x = -100:2:100; y = x; % 设置x和y的范围为[-100, 100]
        case 'F7' 
            x = -1:0.03:1; y = x; % 设置x和y的范围为[-1, 1]
        case 'F8' 
            x = -500:10:500; y = x; % 设置x和y的范围为[-500, 500]
        case 'F9' 
            x = -5:0.1:5; y = x; % 设置x和y的范围为[-5, 5]
        case 'F10' 
            x = -20:0.5:20; y = x; % 设置x和y的范围为[-20, 20]
        case 'F11' 
            x = -500:10:500; y = x; % 设置x和y的范围为[-500, 500]
        case 'F12' 
            x = -10:0.1:10; y = x; % 设置x和y的范围为[-10, 10]
        case 'F13' 
            x = -5:0.08:5; y = x; % 设置x和y的范围为[-5, 5]
        case 'F14' 
            x = -100:2:100; y = x; % 设置x和y的范围为[-100, 100]
        case 'F15' 
            x = -5:0.1:5; y = x; % 设置x和y的范围为[-5, 5]
        case 'F16' 
            x = -1:0.01:1; y = x; % 设置x和y的范围为[-1, 1]
        case 'F17' 
            x = -5:0.1:5; y = x; % 设置x和y的范围为[-5, 5]
        case 'F18' 
            x = -5:0.06:5; y = x; % 设置x和y的范围为[-5, 5]
        case 'F19' 
            x = -5:0.1:5; y = x; % 设置x和y的范围为[-5, 5]
        case 'F20' 
            x = -5:0.1:5; y = x; % 设置x和y的范围为[-5, 5]
        case 'F21' 
            x = -5:0.1:5; y = x; % 设置x和y的范围为[-5, 5]
        case 'F22' 
            x = -5:0.1:5; y = x; % 设置x和y的范围为[-5, 5]
        case 'F23' 
            x = -5:0.1:5; y = x; % 设置x和y的范围为[-5, 5]
    end

    % 获取绘图点的数量
    L = length(x);
    f = []; % 初始化函数值矩阵

    % 计算每个点的函数值
    for i = 1:L
        for j = 1:L
            if strcmp(func_name, 'F15') == 0 && strcmp(func_name, 'F19') == 0 && strcmp(func_name, 'F20') == 0 && strcmp(func_name, 'F21') == 0 && strcmp(func_name, 'F22') == 0 && strcmp(func_name, 'F23') == 0
                f(i, j) = fobj([x(i), y(j)]); % 计算函数值
            end
            if strcmp(func_name, 'F15') == 1
                f(i, j) = fobj([x(i), y(j), 0, 0]); % 计算函数值
            end
            if strcmp(func_name, 'F19') == 1
                f(i, j) = fobj([x(i), y(j), 0]); % 计算函数值
            end
            if strcmp(func_name, 'F20') == 1
                f(i, j) = fobj([x(i), y(j), 0, 0, 0, 0]); % 计算函数值
            end       
            if strcmp(func_name, 'F21') == 1 || strcmp(func_name, 'F22') == 1 || strcmp(func_name, 'F23') == 1
                f(i, j) = fobj([x(i), y(j), 0, 0]); % 计算函数值
            end          
        end
    end

    % 绘制函数的三维曲面图
    surfc(x, y, f, 'LineStyle', 'none');

end

initialization

%_________________________________________________________________________%
%  Whale Optimization Algorithm (WOA) source codes demo 1.0               %
%                                                                         %
%  Developed in MATLAB R2011b(7.13)                                       %
%                                                                         %
%  Author and programmer: Seyedali Mirjalili                              %
%                                                                         %
%         e-Mail: ali.mirjalili@gmail.com                                 %
%                 seyedali.mirjalili@griffithuni.edu.au                   %
%                                                                         %
%       Homepage: http://www.alimirjalili.com                             %
%                                                                         %
%   Main paper: S. Mirjalili, A. Lewis                                    %
%               The Whale Optimization Algorithm,                         %
%               Advances in Engineering Software , in press,              %
%               DOI: http://dx.doi.org/10.1016/j.advengsoft.2016.01.008   %
%                                                                         %
%_________________________________________________________________________%

% 此函数初始化第一批搜索代理的位置
function Positions = initialization(SearchAgents_no, dim, ub, lb)

Boundary_no = size(ub, 2); % 获取上边界的列数，即变量的数量

% 如果所有变量的边界相同且用户输入单个数值作为上下边界
if Boundary_no == 1
    Positions = rand(SearchAgents_no, dim) .* (ub - lb) + lb; % 生成在上下边界之间的随机数
end

% 如果每个变量有不同的上下边界
if Boundary_no > 1
    for i = 1:dim % 遍历每个维度
        ub_i = ub(i); % 获取第i个维度的上边界
        lb_i = lb(i); % 获取第i个维度的下边界
        Positions(:, i) = rand(SearchAgents_no, 1) .* (ub_i - lb_i) + lb_i; % 生成在上下边界之间的随机数
    end
end