Create svm_sklearn.py

Author: ujjwalkarn

svm_sklearn.py

@@ -0,0 +1,15 @@
+import numpy as np
+from sklearn import svm
+
+# Read the data
+train = np.loadtxt(open("train.csv","rb"), delimiter=",", skiprows=0)
+trainLabels = np.loadtxt(open("trainLabels.csv","rb"), delimiter=",", skiprows=0)
+test = np.loadtxt(open("test.csv","rb"), delimiter=",", skiprows=0)
+
+
+X, y = train, trainLabels
+s = svm.SVC()
+s.fit(X, y)
+
+predictions = s.predict(test)
+np.savetxt("fancySVMSubmission.csv", predictions.astype(int), fmt='%d', delimiter=",")