mshenoda commited on
Commit
0c28a58
1 Parent(s): ce46de0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -0
README.md CHANGED
@@ -24,6 +24,11 @@ language:
24
  # RoBERTa based Spam Message Detection
25
  Spam messages frequently carry malicious links or phishing attempts posing significant threats to both organizations and their users. By choosing our RoBERTa-based spam message detection system, organizations can greatly enhance their security infrastructure. Our system effectively detects and filters out spam messages, adding an extra layer of security that safeguards organizations against potential financial losses, legal consequences, and reputational harm.
26
 
 
 
 
 
 
27
  ## Model Output
28
  - 0 is ham
29
  - 1 is spam
 
24
  # RoBERTa based Spam Message Detection
25
  Spam messages frequently carry malicious links or phishing attempts posing significant threats to both organizations and their users. By choosing our RoBERTa-based spam message detection system, organizations can greatly enhance their security infrastructure. Our system effectively detects and filters out spam messages, adding an extra layer of security that safeguards organizations against potential financial losses, legal consequences, and reputational harm.
26
 
27
+ ## Metrics
28
+ Loss | Accuracy | Precision / Recall | Confusion Matrix
29
+ :-------------------------:|:-------------------------:|:-------------------------:|:-------------------------:
30
+ ![](plots/train_validation_loss.jpg "Train / Validation Loss") Train / Validation | ![](plots/validation_accuracy.jpg "Validation Accuracy") Validation | ![](plots/validation_precision_recall.jpg "Validation Precision / Recall") Validation | ![](plots/confusion_matrix.png "confusion_matrix") Testing Set
31
+
32
  ## Model Output
33
  - 0 is ham
34
  - 1 is spam