diff --git a/naive_bayes/nb_author_id.py b/naive_bayes/nb_author_id.py index f69d57d8408..fd471fd519d 100644 --- a/naive_bayes/nb_author_id.py +++ b/naive_bayes/nb_author_id.py @@ -27,7 +27,17 @@ ######################################################### ### your code goes here ### - +from sklearn.naive_bayes import GaussianNB +clf = GaussianNB() +t0 = time() +clf.fit(features_train, labels_train) +print "training time:", round(time()-t0, 3), "s" +t0 = time() +pred = clf.predict(features_test) +print "training time:", round(time()-t0, 3), "s" + +from sklearn.metrics import accuracy_score +print accuracy_score(pred, labels_test) #########################################################