文章目录粒子群算法(PSO)代码实现1、导入需要的库2、设置参数3、适应度函数4、初始化粒子群5、迭代更新粒子群
粒子群算法(PSO)
PSO 通过模拟鸟群的捕食行为来求取最优解。
假设一群鸟在随机搜索食物。在这个区域里只有一块食物(对应着最优解)。所有的鸟都不知道食物的具体位置,但是它们可以判断自身与食物的大致距离,即通过 fit 值判断与最优解的距离。那么找到食物的最优策略就是搜寻目前离食物最近的鸟的周围区域。
PSO 中,问题的每个解都是搜索空间中的一只“鸟”。我们称之为“粒子”。所有的粒子都有一个由被优化的函数决定的适应值,并且所有的粒子都具有速度和位置两个属性。
在每一次迭代中,粒子通过跟踪两个”极值”来更新自己。第一个极值就是粒子本身所找到的最优解 p_best。另一个极值是整个种群目前找到的最优解,即全局极值 g_best。
粒子群算法的求解伪代码如下:
初始化粒子群
while 未达到最大迭代次数或最小损失:
for 粒子 in 粒子群:
计算粒子适应度
if 此粒子适应度高于此粒子在所有迭代次数的最佳适应度 p_bestfit:
将该值设置为新的 p_bestfit
if 所有粒子中的最佳适应度大于整个粒子群在所有迭代次数的最佳适应度 p_bestfit:
选择有最佳适应度的粒子作为 g_best,将该适应度设置为新的 g_bestfit
for 粒子 in 粒子群:
计算粒子速度
更新粒子位置
其中,粒子速度的更新公式为:
其中:
代表利用粒子本身所找到的最优解更新自己的速度。
代表利用整个种群目前找到的最优解更新自己的速度。
粒子位置的更新公式为:
在以上公式中,v[i]v[i]v[i] 是第 iii 个粒子的速度,www 是惯性权重(有助于跳出局部最优解),present[i]present[i]present[i] 是第 iii 个粒子的当前位置,pbest[i]pbest[i]pbest[i] 是第 iii 个粒子的历史最佳,gbestgbestgbest 是全局最佳,rand()rand ()rand() 是介于 0 到 1 之间的随机数。c1、c2c1、c2c1、c2 是学习因子,通常情况下,c1=c2=2c1 = c2 = 2c1=c2=2。
代码实现
1、导入需要的库