diff --git a/svm/svm_author_id.py b/svm/svm_author_id.py index fda3f7fdb28..6159ebf6cfe 100644 --- a/svm/svm_author_id.py +++ b/svm/svm_author_id.py @@ -24,6 +24,14 @@ ######################################################### ### your code goes here ### +import sklearn.svm import SVC +clf = SVC() #Default params here +clf.fit(features_train, features_test) +yhat = clf.predict(labels_train) + +from sklearn.metrics import accuracy_score +acc = accuracy_score(yhat, labels_test) +print(acc) #########################################################