Adj

tulane-cmps6730 · Apr 28, 2024 · e06d6e6 · e06d6e6
1 parent ab18fb2
commit e06d6e6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nlp/cli.py b/nlp/cli.py
@@ -102,7 +102,7 @@ def process_text(document):
     X = vec_1.fit_transform(train_df["Comment"])
     y = train_df["Result_Bin"]
     bnb.fit(X,y)
-    y_pred = nb.predict(val_df["Stemmed"])
+    y_pred = bnb.predict(val_df["Stemmed"])
     y_val = val_df["Result_Bin"]
     # Calculate F1
     f1 = f1_score(y_val, y_pred)