update

Author: bob7783

nlp_class/spam2.py

@@ -54,7 +54,7 @@
 model.fit(Xtrain, Ytrain)
 print("train score:", model.score(Xtrain, Ytrain))
 print("test score:", model.score(Xtest, Ytest))
-exit()
+# exit()
 
 
 # visualize the data
@@ -73,6 +73,7 @@ def visualize(label):
 
 
 # see what we're getting wrong
+X = tfidf.transform(df['data'])
 df['predictions'] = model.predict(X)
 
 # things that should be spam