End of training

README.md

@@ -5,6 +5,9 @@ tags:
 - generated_from_trainer
 metrics:
 - accuracy
+- precision
+- recall
+- f1
 model-index:
 - name: distilbert-scam-classifier-v1
   results: []
@@ -17,8 +20,11 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0051
-- Accuracy: 1.0
+- Loss: 0.0039
+- Accuracy: {'accuracy': 1.0}
+- Precision: {'precision': 1.0}
+- Recall: {'recall': 1.0}
+- F1: {'f1': 1.0}
 
 ## Model description
 
@@ -47,12 +53,12 @@ The following hyperparameters were used during training:
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss | Accuracy |
-|:-------------:|:-----:|:----:|:---------------:|:--------:|
-| No log        | 1.0   | 40   | 0.0998          | 1.0      |
-| No log        | 2.0   | 80   | 0.0134          | 1.0      |
-| No log        | 3.0   | 120  | 0.0061          | 1.0      |
-| No log        | 4.0   | 160  | 0.0051          | 1.0      |
+| Training Loss | Epoch | Step | Validation Loss | Accuracy          | Precision          | Recall          | F1          |
+|:-------------:|:-----:|:----:|:---------------:|:-----------------:|:------------------:|:---------------:|:-----------:|
+| No log        | 1.0   | 40   | 0.0561          | {'accuracy': 1.0} | {'precision': 1.0} | {'recall': 1.0} | {'f1': 1.0} |
+| No log        | 2.0   | 80   | 0.0150          | {'accuracy': 1.0} | {'precision': 1.0} | {'recall': 1.0} | {'f1': 1.0} |
+| No log        | 3.0   | 120  | 0.0043          | {'accuracy': 1.0} | {'precision': 1.0} | {'recall': 1.0} | {'f1': 1.0} |
+| No log        | 4.0   | 160  | 0.0039          | {'accuracy': 1.0} | {'precision': 1.0} | {'recall': 1.0} | {'f1': 1.0} |
 
 
 ### Framework versions