1

tulane-cmps6730 · Apr 29, 2024 · ad7791c · ad7791c
1 parent f9d6cec
commit ad7791c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/nlp/app/forms.py b/nlp/app/forms.py
@@ -12,6 +12,7 @@ class Meta:  # Ignoring CSRF security feature.
 	model_choice = RadioField('Choose Model', choices=[
 		('bnb', 'Naive Bayes'), 
 		('lr', 'Logistic Regression'),
-		('cnn', 'CNN')
+		('cnn', 'CNN'),
+        ('bert', "BERT")
 	])
 	submit = SubmitField('Submit')
diff --git a/nlp/cli.py b/nlp/cli.py
@@ -242,9 +242,9 @@ def __len__(self):
     precision = precision_score(true_labels, predictions)
     recall = recall_score(true_labels, predictions)
     f1 = f1_score(true_labels, predictions)
-    print("Precision on Test:", precision)
-    print("Recall on Test:", recall)
-    print("F1 Score on Test:", f1)
+    print("Precision on Test:", round(precision,3))
+    print("Recall on Test:", round(recall,3))
+    print("F1 Score on Test:", round(f1,3))
 
 if __name__ == "__main__":
     sys.exit(main())