Python代码下的PSO实现及Matlab下的粒子群函数分享

以ras函数(rastrigin's function)为目标函数,求其在x1,x2∈[-5,5]上的最小值。这个函数对模拟退火、进化计算等算法具有很强的欺骗性,因为它有非常多的局部最小值点和局部最大值点,很容易使算法陷入局部最优,而不能得到全局最优解。如下图所示,该函数只在(0,0)处存在全局最小值0。
python代码实现
import numpy as npimport matplotlib.pyplot as plt# 目标函数定义def ras(x): y = 20 + x[0] ** 2 + x[1] ** 2 - 10 * (np.cos(2 * np.pi * x[0]) + np.cos(2 * np.pi * x[1])) return y# 参数初始化w = 1.0c1 = 1.49445c2 = 1.49445maxgen = 200 # 进化次数sizepop = 20 # 种群规模# 粒子速度和位置的范围vmax = 1vmin = -1popmax = 5popmin = -5# 产生初始粒子和速度pop = 5 * np.random.uniform(-1, 1, (2, sizepop))v = np.random.uniform(-1, 1, (2, sizepop))fitness = ras(pop) # 计算适应度i = np.argmin(fitness) # 找最好的个体gbest = pop # 记录个体最优位置zbest = pop[:, i] # 记录群体最优位置fitnessgbest = fitness # 个体最佳适应度值fitnesszbest = fitness[i] # 全局最佳适应度值# 迭代寻优t = 0record = np.zeros(maxgen)while t vmax] = vmax # 限制速度 v[v popmax] = popmax # 限制位置 pop[pop 0.8: # 如果这个数落在变异概率区间内,则进行变异处理 k = np.random.randint(0,2) # 在[0,2)之间随机选一个整数 pop[:,k] = np.random.random() # 在选定的位置进行变异 ''' # 计算适应度值 fitness = ras(pop) # 个体最优位置更新 index = fitness < fitnessgbest fitnessgbest[index] = fitness[index] gbest[:, index] = pop[:, index] # 群体最优更新 j = np.argmin(fitness) if fitness[j] < fitnesszbest: zbest = pop[:, j] fitnesszbest = fitness[j] record[t] = fitnesszbest # 记录群体最优位置的变化 t = t + 1# 结果分析print(zbest)plt.plot(record, 'b-')plt.xlabel('generation')plt.ylabel('fitness')plt.title('fitness curve')plt.show()结果为
[0.99699579 0.00148844]
可以知道求解的点非最小值,算法陷入了局部最小值。
删除自适应变异部分的注释,运行后结果如下,可以看出收敛到全局最优解。
[0.00022989 0.00014612]
matlab有个自带的粒子群优化函数particleswarm也可以使用。本例的代码如下:
y = @(x) 20 + x(1).^2 + x(2).^2 - 10*(cos(2*pi*x(1))+cos(2*pi*x(2)));rng defaultoptions = optimoptions('particleswarm','swarmsize',200,'hybridfcn',@fmincon,'maxiterations',200, 'display','iter');lb = [-5 -5]; % 这是变量的下限ub = [5 5]; % 这是变量的上限[x,fval,exitflag,output] = particleswarm(y,length(lb),lb,ub,options);结果如下

影响马达性能的线圈 (上)
iPhone12 mini因销量不佳将在第二季度停产
镍镉电池充不进电怎么解决
如何利用先进科技技术来保障网络信息安全
人体识别图像技术在智能安防中的应用
Python代码下的PSO实现及Matlab下的粒子群函数分享
还有这种操作?给CPU换个纯铜镜面IHS,温度还可以进一步下降7°C
放大器的常见故障及预防措施
一文看懂Linux性能分析之perf原理
MAX9272 28位GMSL解串器为同轴电缆或STP电缆
关于接地电阻柜面板装配工艺
小米究竟该不该为经小米导流投资P2P而踩雷的“米粉”负责呢
回流焊的工艺流程及工艺特点
小型降压转换器电路的设计
Qualcomm联手京东方开发3D Sonic传感器相关产品
探讨技术的跨界融合与商业模式的创新实践
ESP8266二级bootload跳转测试
了解物联网部署的总拥有成本,以优化支出
Flash存储器闪存工作原理及具体步骤
皇冠假日酒店正在试验昼夜节律照明,以帮助其住客更好地睡觉