Create svm_sklearn.py

ds-ebooks · May 16, 2016 · f2a2593 · f2a2593
1 parent 384c236
commit f2a2593
Showing 1 changed file with 15 additions and 0 deletions.
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=",")