test_func.m

来自「PSO工具箱使用简介 方便大家 共同进步」· M 代码 · 共 20 行

% 适用于z=f(x,y)的类型，计算z的最小/最大值
% 例如计算0.5*(x-3)^2+0.2*(y-5)^2-0.1的最小值, x: -50:50; y: -50:50.
%%--------------------------------------------------------------------%%
function z=test_func(in)

   nn=size(in);  %输入的是矩阵，即算法中随机产生一组x和y，按 [x(nn,1), y(nn,1)]排列
   x=in(:,1);
   y=in(:,2);
   nx=nn(1);

   for i=1:nx
       %temp = 0.5*(x(i)-3)^2+sin(y(i))-0.1;
       temp = 0.5*(x(i)-3)^2+0.2*(y(i)-5)^2-0.1;
       z(i,:) = temp;
   end
%%--------------------------------------------------------------------%%