MIN Better str representation

luispedro · Aug 15, 2013 · f3e9348 · f3e9348
1 parent 5db4c6f
commit f3e9348
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/milk/supervised/svm.py b/milk/supervised/svm.py
@@ -154,6 +154,9 @@ def __call__(self, x1, x2):
         d2 = d2.sum()
         res = self.beta*np.exp(-d2/self.sigma)
         return res
+    def __str__(self):
+        return 'rbf_kernel(%s, %s)' % (self.sigma, self.beta)
+    __repr__ = __str__
 
     def preprocess(self, X):
         return preprocessed_rbf_kernel(X, self.sigma, self.beta)