Update support_vector_machines.py

Author: WeiYFan

machine_learning/support_vector_machines.py

@@ -57,8 +57,8 @@ class SVC:
     >>> SVC(kernel="polynomial",degree=None)
     Traceback (most recent call last):
         ...
-    ValueError: degree must be int
-    >>> SVC(kernel="polynomial",degree=-1)
+    ValueError: degree must be float or int
+    >>> SVC(kernel="polynomial",degree=-1.0)
     Traceback (most recent call last):
         ...
     ValueError: degree must be > 0
@@ -70,7 +70,7 @@ def __init__(
         regularization: float = np.inf,
         kernel: str = "linear",
         gamma: float = 0.0,
-        degree: int = 0.0,
+        degree: float=0.0,
         coef0: float = 0.0,
     ) -> None:
         self.regularization = regularization
@@ -93,8 +93,8 @@ def __init__(
         elif kernel == "polynomial":
             if self.degree == 0:
                 raise ValueError("polynomial kernel requires degree")
-            if not isinstance(self.degree, int):
-                raise ValueError("degree must be int")
+            if not isinstance(self.degree, (float, int)) :
+                raise ValueError("degree must be float or int")
             if not self.degree > 0:
                 raise ValueError("degree must be > 0")
             self.kernel = self.__polynomial
@@ -123,6 +123,7 @@ def __rbf(self, vector1: ndarray, vector2: ndarray) -> float:
         """
         return np.exp(-(self.gamma * norm_squared(vector1 - vector2)))
 
+
     def __polynomial(self, vector1: ndarray, vector2: ndarray) -> float:
         """
         Polynomial kernel: (x . y + coef0)^degree