Skip to content

Commit 9d7532a

Browse files
authored
Merge pull request #3 from drkrillo/sequential_minimum_optimization_add_doctests
Add doctest to _rbf Kernel func
2 parents fb58570 + 308fbe4 commit 9d7532a

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

machine_learning/sequential_minimum_optimization.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -471,6 +471,13 @@ def _linear(self, v1, v2):
471471
return np.inner(v1, v2) + self.coef0
472472

473473
def _rbf(self, v1, v2):
474+
"""
475+
>>> from machine_learning.sequential_minimum_optimization import Kernel
476+
>>> kernel = Kernel(kernel='linear')
477+
>>> result = kernel._rbf(np.array([1, 2]), np.array([2, 3]))
478+
>>> round(result, 3)
479+
0.135
480+
"""
474481
return np.exp(-1 * (self.gamma * np.linalg.norm(v1 - v2) ** 2))
475482

476483
def _check(self):

0 commit comments

Comments
 (0)