[feature] default parameters based on hpt

zapata-engineering · Dec 15, 2022 · b18551c · b18551c
1 parent 0cc274d
commit b18551c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/orquestra/opt/optimizers/pso/continuous_pso_optimizer.py b/src/orquestra/opt/optimizers/pso/continuous_pso_optimizer.py
@@ -45,9 +45,9 @@ def __init__(
         self,
         swarm_size: int,
         bounds: Bounds,
-        inertia: float,
-        affinity_towards_best_particle_position: float,
-        affinity_towards_best_swarm_position: float,
+        inertia: float = 0.88,
+        affinity_towards_best_particle_position: float = 0.6,
+        affinity_towards_best_swarm_position: float = 0.26,
         patience: Optional[int] = None,
         delta: float = 1e-10,
         max_iterations: Optional[int] = None,