Swarm

Complexity

Total Complexity

12

Size/Duplication

Total Lines	34
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

5 Methods

Rating	Name	Duplication	Size	Complexity
A	do_single_iteration()	0	8	3
A	__init__()	0	8	2
A	_update_best_particle()	0	5	3
A	_update_velocieties()	0	3	2
A	initialize()	0	4	2

"""Represents swarm."""
import numpy as np

The import numpy could not be resolved.

from grortir.main.pso.particle import Particle


class Swarm(object):
    """Class which represent swarm."""
    def __init__(self, process, stages, number_of_particles):
        self.stages = stages
        self.process = process
        self.number_of_particles = number_of_particles
        self.particles = [Particle(stages, self.process) for i in
                          range(number_of_particles)]
        self.best_particle_quality = np.inf
        self.best_particle = self.particles[0]

    def initialize(self):
        """Initialize all particles in swarm."""
        for particle in self.particles:
            particle.initialize()

    def do_single_iteration(self):
        """Iterate one time."""
        for particle in self.particles:
            particle.update_values()
        self._update_best_particle()
        self._update_velocieties()
        for particle in self.particles:
            particle.move()

    def _update_best_particle(self):

This method should have a docstring.

        for particle in self.particles:
            if particle.best_quality < self.best_particle_quality:
                self.best_particle = particle
                self.best_particle_quality = particle.best_quality

    def _update_velocieties(self):

This method should have a docstring.

        for particle in self.particles:
            particle.update_velocieties(self.best_particle)