chore(opytimizer): Adds Walrus Optimization Algorithm

douglasrodrigues · douglasrodrigues · commit d36166bd5351 · 2023-09-06T20:07:15.000-03:00
diff --git a/examples/optimizers/swarm/create_waoa.py b/examples/optimizers/swarm/create_waoa.py
@@ -0,0 +1,4 @@
+from opytimizer.optimizers.swarm import WAOA
+
+# Creates a WAOA optimizer
+o = WAOA()
diff --git a/opytimizer/optimizers/swarm/__init__.py b/opytimizer/optimizers/swarm/__init__.py
@@ -29,4 +29,5 @@
 from opytimizer.optimizers.swarm.ssa import SSA
 from opytimizer.optimizers.swarm.sso import SSO
 from opytimizer.optimizers.swarm.stoa import STOA
+from opytimizer.optimizers.swarm.waoa import WAOA
 from opytimizer.optimizers.swarm.woa import WOA
diff --git a/opytimizer/optimizers/swarm/waoa.py b/opytimizer/optimizers/swarm/waoa.py
@@ -0,0 +1,66 @@
+"""Walrus Optimization Algorithm.
+"""
+
+import copy
+from typing import Any, Dict, Optional
+
+import numpy as np
+
+import opytimizer.math.random as r
+import opytimizer.utils.logging as l
+from opytimizer.core.optimizer import Optimizer
+from opytimizer.core.space import Space
+
+logger = l.get_logger(__name__)
+
+class WAOA(Optimizer):
+    """A WAOA class, inherited from Optimizer.
+
+    This is the designed class to dife WAOA-related 
+    variables and methods.
+
+    References:
+        P. Trojovsk&yacute; and M. Dehghani. A new bio-inspired metaheuristic algorithm for
+        solving optimization problems based on walruses behavior. Scientific Reports (2023).
+
+    """
+
+    def __init__(self, params: Optional[Dict[str, Any]] = None) -> None:
+        """Initialization method.
+
+        Args:
+            params (str): Contains key-value parameters to the meta-heuristics.
+        """
+
+        logger.info('Overriding class: Optimizer -> SSA')
+        
+        super(WAOA, self).__init__()
+
+        self.build(params)
+
+        logger.info('Class overrided.')
+
+    def update(self, space: Space) -> None:
+        """Wraps Walrus Optimization Algorithm over all agents and variables.
+
+        Args:
+            space: Space containing agents and update-related information.
+
+        """
+
+        for agent in space.agents:
+            a = copy.deepcopy(agent)
+
+            for j in range(space.n_variables):
+                
+                k = r.generate_integer_random_number(1, 2)
+                r1 = r.generate_uniform_random_number()
+
+                a.position[j] = agent.position[j] + r1 * (space.best_agent.position[j] - k * agent.position[j])
+
+            a.clip_by_bound()
+
+            a.fit = function(a.position)
+            if a.fit < agent.fit:
+                agent.position = copy.deepcopy(a.position)
+                agent.fit = copy.deepcopy(a.fit)
diff --git a/tests/opytimizer/optimizers/swarm/test_waoa.py b/tests/opytimizer/optimizers/swarm/test_waoa.py
@@ -0,0 +1,16 @@
+import numpy as np
+
+from opytimizer.optimizers.swarm import waoa
+from opytimizer.spaces import search
+
+def test_waoa_update():
+    def square(x):
+        return np.sum(x**2)
+
+    new_waoa = waoa.WAOA()
+
+    search_space = search.SearchSpace(
+        n_agents=20, n_variables=2, lower_bound=[0, 0], upper_bound=[10, 10]
+    )
+
+    new_waoa.update(search_space, square)
