当前位置：首页 > news >正文

改进的鲸鱼优化算法GSWOA优化神经网络模型BP做二分类和多分类预测模型。程序注释详细，可学习性强

news 2026/3/26 22:09:12

改进的鲸鱼优化算法GSWOA优化神经网络模型BP做二分类和多分类预测模型。程序注释详细，可学习性强。程序语言为matlab。直接替换数据就可以用。程序有混淆矩阵，分类预测图，ROC图。

江湖救急！分类任务总被BP神经网络的参数调优折腾到秃头？今天咱们整点骚操作——用高斯柯西变异鲸鱼算法(GSWOA)给BP神经网络来个全面升级。直接上Matlab代码，手把手教你怎么把二分类、多分类任务按在地上摩擦。

先看核心武器GSWOA的变异操作，这货在传统鲸鱼算法里加了双重保险：

% GSWOA位置更新核心代码 for i=1:SearchAgents if p<0.5 if abs(A)>=1 X_rand = X_rand_array(i,:); D = abs(C*X_rand - X(i,:)); X_new = X_rand - A*D; else D = abs(C*X(i_best,:) - X(i,:)); X_new = X(i_best,:) - A*D; end else D_best = abs(X(i_best,:) - X(i,:)); X_new = D_best*exp(b.*l).*cos(2*pi*l) + X(i_best,:); end % 高斯-柯西联合变异 if rand >0.8 sigma = 0.1*(ub-lb); X_new = X_new + sigma.*randn(size(X_new)).*(1 - (iter/Max_iter)) + ... (ub-lb).*tan(pi*(rand-0.5))*(iter/Max_iter); end end

这段代码的骚气在于：迭代前期用柯西变异扩大搜索范围，后期用高斯变异精细搜索，自适应权重让算法在全局探索和局部开发之间丝滑切换。

网络训练部分咱们整了个三明治结构：

% 创建BP网络 net = feedforwardnet(hiddenLayerSize); net.layers{1}.transferFcn = 'tansig'; % 隐藏层激活函数 net.layers{2}.transferFcn = 'softmax'; % 多分类输出用softmax net.trainFcn = 'trainlm'; % 默认Levenberg-Marquardt算法 % 设置超参数搜索范围 dim = inputnum*hiddenLayerSize + hiddenLayerSize + ... hiddenLayerSize*outputnum + outputnum; % 权重阈值总数 lb = -3*ones(1,dim); % 参数下界 ub = 3*ones(1,dim); % 参数上界

这里暗藏玄机：inputnum和outputnum根据数据自动适配，二分类用sigmoid，多分类切softmax，一个代码通吃所有分类任务。

结果可视化才是装逼的正确姿势：

% ROC曲线绘制 figure('Color',[1 1 1]) plot(roc_x, roc_y, 'b', 'LineWidth', 1.5) hold on plot([0 1], [0 1], 'k--') xlabel('False positive rate') ylabel('True positive rate') title(['ROC曲线 (AUC = ' num2str(auc) ')']) legend('预测结果', '随机猜测', 'Location', 'Best') set(gca, 'FontSize', 12)

这个ROC画法暗藏彩蛋：自动计算AUC面积，虚线参考线一眼看出模型提升幅度。混淆矩阵更带感：

% 混淆矩阵增强版 cm = confusionchart(real_labels, predict_labels); cm.Title = sprintf('分类准确率: %.2f%%', 100*accuracy); cm.FontSize = 12; cm.DiagonalColor = [0.2 0.8 0.2]; cm.OffDiagonalColor = [0.8 0.2 0.2];

颜色区分正确错误分类，准确率直接标在标题，论文级逼格分分钟搞定。

实战建议：