chore: add docstring

CogitoNTNU · Nov 3, 2024 · a79f5a4 · a79f5a4
1 parent f1288c0
commit a79f5a4
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/NEATnetwork.py b/src/NEATnetwork.py
@@ -18,6 +18,14 @@ def __init__(self, genome: Genome):
         self.topological_order = self._topological_sort()
 
     def sigmoid(self, x: np.ndarray) -> np.ndarray:
+        """ Sigmoid activation function.
+
+        Args:
+            x (np.ndarray): Input array
+
+        Returns:
+            np.ndarray: Output array after applying sigmoid function
+        """
         return 1 / (1 + np.exp(-x))
 
     def ReLU(self, x):