Add doctest to _rbf Kernel func

TheAlgorithms · Sep 18, 2024 · 308fbe4 · 308fbe4
1 parent 94b1403
commit 308fbe4
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/machine_learning/sequential_minimum_optimization.py b/machine_learning/sequential_minimum_optimization.py
@@ -471,6 +471,13 @@ def _linear(self, v1, v2):
         return np.inner(v1, v2) + self.coef0
 
     def _rbf(self, v1, v2):
+        """
+        >>> from machine_learning.sequential_minimum_optimization import Kernel
+        >>> kernel = Kernel(kernel='linear')
+        >>> result = kernel._rbf(np.array([1, 2]), np.array([2, 3]))
+        >>> round(result, 3)
+        0.135
+        """
         return np.exp(-1 * (self.gamma * np.linalg.norm(v1 - v2) ** 2))
 
     def _check(self):