diff --git a/naive_bayes/nb_author_id.py b/naive_bayes/nb_author_id.py
index f69d57d8408..98d9c8910d6 100644
--- a/naive_bayes/nb_author_id.py
+++ b/naive_bayes/nb_author_id.py
@@ -26,8 +26,13 @@
 
 #########################################################
 ### your code goes here ###
+from sklearn.naive_bayes import GaussianNB
+clf = GaussianNB()
+clf.fit(features_train, features_test)
+yhat = clf.predict(labels_train)
 
-
+from sklearn.metrics import accuracy_score
+print(accuracy_score(yhat, labels_test))
 #########################################################