粒子群算法(PSO)的实现,具有详细的注释
# -*- coding: utf-8 -*-
"""
Created on Mon Mar 14 09:06:35 2022
@author: aa
"""
import numpy as np#调用numpy
import matplotlib.pylab as plt#画图
np.random.seed(1)#确实随机种子
def fitnessEvalute(x,problemIndex):#适应性评估
# x:被优化问题的候选解,可以是矩阵形式,也可是一维数组形式。
# problemIndex: 被优化问题的序号,方便主程序调用。
if problemIndex == 1: # 球形函数
return np.sum(x**2,axis=1)
elif problemIndex == 2:# Rosenbrock function
共有 0 条评论