diff --git a/svm_class/svm_gradient.py b/svm_class/svm_gradient.py index 1a978363..d4ddf6dc 100644 --- a/svm_class/svm_gradient.py +++ b/svm_class/svm_gradient.py @@ -70,7 +70,7 @@ def fit(self, X, Y, lr=1e-5, n_iters=400): self.alphas[self.alphas < 0] = 0 self.alphas[self.alphas > self.C] = self.C - # distrbution of bs + # distribution of bs idx = np.where((self.alphas) > 0 & (self.alphas < self.C))[0] bs = Y[idx] - (self.alphas * Y).dot(self.kernel(X, X[idx])) self.b = np.mean(bs)