Add doctest for Kernet _linear func

TheAlgorithms · Sep 18, 2024 · 94b1403 · 94b1403
1 parent 8d4c9e8
commit 94b1403
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
@@ -461,6 +461,13 @@ def _polynomial(self, v1, v2):
         return (self.gamma * np.inner(v1, v2) + self.coef0) ** self.degree
 
     def _linear(self, v1, v2):
+        """
+        >>> from machine_learning.sequential_minimum_optimization import Kernel
+        >>> kernel = Kernel(kernel='linear')
+        >>> result = kernel._polynomial(np.array([1, 2]), np.array([2, 3]))
+        >>> int(result) == 8
+        True
+        """
         return np.inner(v1, v2) + self.coef0
 
     def _rbf(self, v1, v2):