matlab调用粒子群函数做最小二乘拟合代码

以下是使用粒子群算法进行最小二乘拟合的MATLAB代码：

function [x, yfit] = psolm(xdata, ydata, func, n, m, lb, ub, options)

% xdata: 自变量数据，1xn向量
% ydata: 因变量数据，1xn向量
% func: 拟合函数
% n: 待拟合参数个数
% m: 粒子数
% lb: 参数下限向量，1xn向量
% ub: 参数上限向量，1xn向量
% options: 优化参数

if nargin < 8
    options = optimoptions('particleswarm', 'MaxIterations', 200, 'Display', 'off');
end

% 定义目标函数
fun = @(x) sum((ydata - func(xdata, x)).^2);

% 粒子群算法优化
[x, ~] = particleswarm(fun, n, lb, ub, options);

% 计算拟合后的因变量数据
yfit = func(xdata, x);

end

使用方法如下：

% 定义拟合函数
func = @(x, p) p(1) * x + p(2);

% 生成测试数据
xdata = 1:10;
ydata = 2 * xdata + 3 + randn(size(xdata));

% 调用粒子群算法拟合
[x, yfit] = psolm(xdata, ydata, func, 2, 50, [-10 -10], [10 10]);

% 绘制拟合结果
plot(xdata, ydata, 'o', xdata, yfit, '-')
legend('原始数据', '拟合结果')