rsarai
diff --git a/‎__pycache__/firefly_algorithm.cpython-35.pyc
-3.65 KB b/‎__pycache__/firefly_algorithm.cpython-35.pyc
-3.65 KB
diff --git a/‎__pycache__/fish.cpython-35.pyc
-1.51 KB b/‎__pycache__/fish.cpython-35.pyc
-1.51 KB
diff --git a/‎__pycache__/fish_school_search.cpython-35.pyc
-4.01 KB b/‎__pycache__/fish_school_search.cpython-35.pyc
-4.01 KB
diff --git a/‎__pycache__/functions.cpython-35.pyc
-2.19 KB b/‎__pycache__/functions.cpython-35.pyc
-2.19 KB
diff --git a/‎__pycache__/parameters.cpython-35.pyc
-270 Bytes b/‎__pycache__/parameters.cpython-35.pyc
-270 Bytes
diff --git a/‎__pycache__/particle.cpython-35.pyc
-2.53 KB b/‎__pycache__/particle.cpython-35.pyc
-2.53 KB
diff --git a/‎fish_school_search.py
Lines changed: 3 additions & 2 deletions b/‎fish_school_search.py
Lines changed: 3 additions & 2 deletions
@@ -2,6 +2,7 @@
 import functools
 import random
 import math
+import numpy as np
 
 from particle import Fish
 from parameters import num_of_individuos, dimensions, iterations_number
@@ -69,7 +70,7 @@ def search(self):
             # calculates the baricenter
             weights_sum = 0
             for fish in  self.population:
-                weights_sum = fish.weights
+                weights_sum += fish.weights
 
             weights_sum_times_position = 0
             for fish in self.population:
@@ -80,7 +81,7 @@ def search(self):
 
             # executes volitive moviment
             for fish in self.population:
-                distance_to_baricenter = self._euclidian_distance(fish.current_position, baricenter)
+                distance_to_baricenter = np.linalg.norm([fish.current_position, baricenter])
 
                 volational_array = []
                 difference_to_baricenter = []