going with kernels

LordSomen · Jul 15, 2018 · 8f6838e · 8f6838e
1 parent 814400c
commit 8f6838e
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/SVM/svm_model.py b/SVM/svm_model.py
@@ -31,6 +31,9 @@
 print(Y)
 
 #%%
+'''
+for non linear distinction datasets 
+'''
 polynomial_svm_clf = Pipeline((
     ("poly_features",PolynomialFeatures()),
     ('scaler',StandardScaler()),
@@ -45,4 +48,12 @@
 ("scaler", StandardScaler()),
 ("svm_clf", SVC(kernel="poly", degree=3, coef0=1, C=5))
 ))
-poly_kernel_svm_clf.fit(X, Y)
+poly_kernel_svm_clf.fit(X, Y)
+
+#%%
+rbf_kernel_svm_clf = Pipeline((
+    ("scaler",StandardScaler()),
+    ("svm_clf",SVC(kernel='rbf',gamma=5,C=0.001))
+))
+rbf_kernel_svm_clf.fit(X,Y)
+