今天给大家带来一期由改进的北方苍鹰算法(SCNGO)优化VMD的两个参数。

同样以西储大学数据集为例，选用105.mat中的X105_BA_time.mat数据中1000个数据点。没有数据的看这篇文章。西储大学轴承诊断数据处理，matlab免费代码获取

选取四种适应度函数进行优化，以此确定VMD的最佳k和α参数。四种适应度函数分别是：最小包络熵，最小样本熵，最小信息熵，最小排列熵。代码中可以一键切换。至于应该选择哪种作为自己的适应度函数，大家可以看这篇文章。VMD为什么需要进行参数优化，最小包络熵/样本熵/排列熵/信息熵，适应度函数到底该选哪个

1.最小包络熵作为适应度函数

   

2.最小样本熵作为适应度函数

3.最小信息熵作为适应度函数

4.最小排列熵作为适应度函数

代码：

%% 以最小包络熵、最小样本熵、最小信息熵、最小排列熵为目标函数（任选其一），采用SCNGO算法优化VMD，求取VMD最佳的两个参数
clear
clc
close all
xz = 3;  %xz=1 or 2, 选择1，以最小包络熵为适应度函数，选择2，以最小样本熵为适应度函数，选择3，以最小信息熵为适应度函数，选择4，以最小排列熵为适应度函数
if xz == 1  
    fobj=@EnvelopeEntropyCost;          %最小包络熵
elseif xz == 2
    fobj=@SampleEntropyCost;            %最小样本熵
elseif xz == 3
    fobj=@infoEntropyCost;              %最小信息熵
elseif xz == 4
    fobj=@PermutationEntropyCost;       %最小排列熵
end


%% 选取数据
load 105.mat
da = X105_DE_time(6001:7000); %这里选取105的DEtime数据，注意这里替换为自己的数据即可，数据形式为n行*1列，列数必须为1。


%% 设置参数
lb = [100 3];    %惩罚因子和K的下限
ub = [2500 10];  %惩罚因子和K的上限
dim = 2;            % 优化变量数目
Max_iter=20;       % 最大迭代数目
SearchAgents_no=20;       %种群规模


%% 调用SCNGO函数
[fMin , bestX, Convergence_curve ] = SCNGO(SearchAgents_no,Max_iter,lb,ub,dim,fobj,da);


%% 画适应度函数曲线图，并输出最佳参数
figure
plot(Convergence_curve,'Color',[0.9 0.5 0.1],'Marker','>','LineStyle','--','linewidth',1);
title('Objective space')
xlabel('Iteration');
ylabel('Best score obtained so far');
legend('SCNGO优化VMD')
display(['The best solution obtained by SCNGO is : ', num2str(round(bestX))]);  %输出最佳位置
display(['The best optimal value of the objective funciton found by SCNGO is : ', num2str(fMin)]);  %输出最佳适应度值

大家注意看到xz这个变量，当选择1，以最小包络熵为适应度函数，选择2，以最小样本熵为适应度函数，选择3，以最小信息熵为适应度函数，选择4，以最小排列熵为适应度函数。这样大家切换起来就很方便了。

代码附带SCNGO在CEC2005函数的测试。

完整代码获取，后台回复关键词：

SCNGOVMD