class CSTestCase(TestCase):
def setUp(self):
self.pso_custom = ParticleSwarmAlgorithm(NP=40,
D=40,
nFES=1000,
C1=2.0,
C2=2.0,
w=0.7,
vMin=-4,
vMax=4,
benchmark=MyBenchmark())
self.pso_griewank = ParticleSwarmAlgorithm(NP=40,
D=40,
nFES=1000,
C1=2.0,
C2=2.0,
w=0.7,
vMin=-4,
vMax=4,
benchmark='griewank')
def test_custom_works_fine(self):
self.assertTrue(self.pso_custom.run())
def test_griewank_works_fine(self):
self.assertTrue(self.pso_griewank.run())
class CSTestCase(TestCase):
def setUp(self):
self.pso_custom = ParticleSwarmAlgorithm(40, 40, 1000, 2.0, 2.0, 0.7,
-4, 4, MyBenchmark())
self.pso_griewank = ParticleSwarmAlgorithm(40, 40, 1000, 2.0, 2.0, 0.7,
-4, 4, 'griewank')
def test_custom_works_fine(self):
self.assertTrue(self.pso_custom.run())
def test_griewank_works_fine(self):
self.assertTrue(self.pso_griewank.run())
def optimize(bench, algo):
average_mfo = 0
average_de = 0
average_abc = 0
average_pso = 0
average_ba = 0
average_fa = 0
average_ga = 0
for i in np.arange(epoch):
mfo = MothFlameOptimizer(D=dim, NP=pop, nGEN=maxIter, benchmark=bench)
de = DifferentialEvolution(D=dim,
NP=pop,
nGEN=maxIter,
benchmark=bench)
abc = ArtificialBeeColonyAlgorithm(D=dim,
NP=pop,
nFES=maxIter,
benchmark=bench)
pso = ParticleSwarmAlgorithm(D=dim,
NP=pop,
nGEN=maxIter,
benchmark=bench)
ba = BatAlgorithm(D=dim, NP=pop, nFES=maxIter, benchmark=bench)
fa = FireflyAlgorithm(D=dim, NP=pop, nFES=maxIter, benchmark=bench)
ga = GeneticAlgorithm(D=dim, NP=pop, nFES=maxIter, benchmark=bench)
gen, best_de = de.run()
gen, best_mfo = mfo.run()
gen, best_abc = abc.run()
gen, best_pso = pso.run()
gen, best_ba = ba.run()
gen, best_fa = fa.run()
gen, best_ga = ga.run()
average_mfo += best_de / epoch
average_de += best_mfo / epoch
average_abc += best_abc / epoch
average_pso += best_pso / epoch
average_ba += best_ba / epoch
average_fa += best_fa / epoch
average_ga += best_ga / epoch
print(algo, ': DE Average of Bests over', epoch, 'run: ', average_de)
print(algo, ': MFO Average of Bests over', epoch, 'run: ', average_mfo)
print(algo, ': ABC Average of Bests over', epoch, 'run: ', average_abc)
print(algo, ': PSO Average of Bests over', epoch, 'run: ', average_pso)
print(algo, ': BA Average of Bests over', epoch, 'run: ', average_ba)
print(algo, ': FA Average of Bests over', epoch, 'run: ', average_fa)
print(algo, ': GA Average of Bests over', epoch, 'run: ', average_ga)
return [
average_de, average_mfo, average_abc, average_pso, average_ba,
average_fa, average_ga
]
def plot_example():
task = TaskConvPlot(D=50, nFES=50000, nGEN=10000, benchmark=MyBenchmark())
algo = ParticleSwarmAlgorithm(NP=50,
C1=2.0,
C2=2.0,
w=0.5,
vMin=-5,
vMax=5,
task=task)
best = algo.run()
logger.info('%s %s' % (best[0], best[1]))
input('Press [enter] to continue')
def logging_example():
task = TaskConvPrint(D=50, nFES=50000, nGEN=50000, benchmark=MyBenchmark())
algo = ParticleSwarmAlgorithm(NP=50,
C1=2.0,
C2=2.0,
w=0.5,
vMin=-5,
vMax=5,
seed=None,
task=task)
best = algo.run()
logger.info('%s %s' % (best[0], best[1]))
def simple_example(runs=10):
for i in range(10):
algo = ParticleSwarmAlgorithm(NP=50,
D=40,
nFES=40000,
C1=2.0,
C2=2.0,
w=0.5,
vMin=-5,
vMax=5,
seed=i,
benchmark=MyBenchmark())
Best = algo.run()
logger.info('%s %s' % (Best[0], Best[1]))
def executePSO(typeBenchmark):
task = StoppingTask(
D=dimensoes,
nFES=numeroAvaliacoes,
optType=tipoOtimizacao,
benchmark=typeBenchmark,
)
algo = ParticleSwarmAlgorithm(
NP=tamanhoPopulacao,
C1=componenteCognitivo,
C2=componenteSocial,
w=pesoInercial,
vMin=velocidadeMinima,
vMax=velocidadeMaxima,
)
best = algo.run(task=task)
return [task, best[1], best[0]]
# encoding=utf8
# This is temporary fix to import module from parent folder
# It will be removed when package is published on PyPI
import sys
sys.path.append('../')
# End of fix
import random
from NiaPy.algorithms.basic import ParticleSwarmAlgorithm
from NiaPy.task import StoppingTask, OptimizationType
from NiaPy.benchmarks import Sphere
#we will run ParticleSwarmAlgorithm for 5 independent runs
for i in range(5):
task = StoppingTask(D=10, nFES=1000, optType=OptimizationType.MAXIMIZATION, benchmark=Sphere())
algo = ParticleSwarmAlgorithm(NP=40, C1=2.0, C2=2.0, w=0.7, vMin=-4, vMax=4)
best = algo.run(task=task)
print best
logging.basicConfig()
logger = logging.getLogger('examples')
logger.setLevel('INFO')
# For reproducive results
random.seed(1234)
class MyBenchmark(object):
def __init__(self):
self.Lower = -11
self.Upper = 11
def function(self):
def evaluate(D, sol):
val = 0.0
for i in range(D):
val = val + sol[i] * sol[i]
return val
return evaluate
for i in range(10):
Algorithm = ParticleSwarmAlgorithm(50, 40, 40000, 2.0, 2.0, 0.5, -5, 5,
MyBenchmark())
Best = Algorithm.run()
logger.info(Best)