粒子群优化算法python程序_粒子群优化算法-python实现

PSOIndividual.py

import numpy as np

import ObjFunction

import copy

class PSOIndividual:

'''

individual of PSO

'''

def __init__(self, vardim, bound):

'''

vardim: dimension of variables

bound: boundaries of variables

'''

self.vardim = vardim

self.bound = bound

self.fitness = 0.

def generate(self):

'''

generate a rondom chromsome

'''

len = self.vardim

rnd = np.random.random(size=len)

self.chrom = np.zeros(len)

self.velocity = np.random.random(size=len)

for i in xrange(0, len):

self.chrom[i] = self.bound[0, i] + \

(self.bound[1, i] - self.bound[0, i]) * rnd[i]

self.bestPosition = np.zeros(len)

self.bestFitness = 0.

def calculateFitness(self):

'''

calculate the fitness of the chromsome

'''

self.fitness = ObjFunction.GrieFunc(

self.vardim, self.chrom, self.bound)

PSO.py

import numpy as np

from PSOIndividual import PSOIndividual

import random

import copy

import matplotlib.pyplot as plt

class ParticleSwarmOptimization:

'''

the class for Particle Swarm Optimization

'''

def __init__(self, sizepop, vardim, bound, MAXGEN, params):

'''

sizepop: population sizepop

vardim: dimension of variables

bound: boundaries of variables

MAXGEN: termination condition