diff --git a/svm_sklearn.py b/svm_sklearn.py new file mode 100644 index 0000000..8f9953f --- /dev/null +++ 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=",")