Create svm_sklearn.py

ujjwalkarn · ujjwalkarn · commit f2a2593b3b98 · 2016-05-16T21:52:30.000+08:00
diff --git a/svm_sklearn.py b/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=",")
