Create ClassifyNB.py

Author: karan1808

ClassifyNB.py

@@ -0,0 +1,13 @@
+def classify(features_train, labels_train):   
+    ### import the sklearn module for GaussianNB
+    ### create classifier
+    ### fit the classifier on the training features and labels
+    ### return the fit classifier
+    
+    
+    ### your code goes here!
+    from sklearn.naive_bayes import GaussianNB
+    clf = GaussianNB()
+    clf.fit(features_train,labels_train)
+    pred = clf.predict(features_train)
+    return clf