better peak extraction

[imago.git] / pso.py

diff --git a/pso.py b/pso.py
index 9149167..c30bcfd 100644 (file)
--- a/pso.py
+++ b/pso.py
@@ -26,7 +26,7 @@ def move(particle, omega, phi_p, phi_g, v_max, global_best, func_d):
 
 def optimize(dimension, boundary, function_d, n_parts, n_turns):
     pool = multiprocessing.Pool(None)
 
 def optimize(dimension, boundary, function_d, n_parts, n_turns):
     pool = multiprocessing.Pool(None)

-    v_max = 30.
+    v_max = boundary

     particles = [particle(dimension, boundary, v_max, function_d)
                  for _ in xrange(n_parts)]
     gl_best = max(particles)
     particles = [particle(dimension, boundary, v_max, function_d)
                  for _ in xrange(n_parts)]
     gl_best = max(particles)