Mahesh9/distil-bert-fintuned-cfpb

README.md

@@ -4,12 +4,13 @@ base_model: distilbert/distilbert-base-uncased
 tags:
 - generated_from_trainer
 metrics:
+- accuracy
 - precision
+- recall
+- f1
 model-index:
 - name: distil-bert-fintuned-cfpb-complaints-data
   results: []
-widget:
-- text: "it is absurd that i have consistently made timely payments for this account and have never been overdue. i kindly request that you promptly update my account to reflect this accurately."
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -17,10 +18,13 @@ should probably proofread and complete it, then remove this comment. -->
 
 # distil-bert-fintuned-cfpb-complaints-data
 
-This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on the Consumer Financial Protection Data(CFPB) dataset.
+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.1898
-- Precision: 0.8651
+- Loss: 0.1782
+- Accuracy: 0.9457
+- Precision: 0.8632
+- Recall: 0.8368
+- F1: 0.8484
 
 ## Model description
 
@@ -49,9 +53,9 @@ The following hyperparameters were used during training:
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss | Precision |
-|:-------------:|:-----:|:----:|:---------------:|:---------:|
-| 0.2162        | 1.0   | 4105 | 0.1898          | 0.8651    |
+| Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision | Recall | F1     |
+|:-------------:|:-----:|:----:|:---------------:|:--------:|:---------:|:------:|:------:|
+| 0.2216        | 1.0   | 4925 | 0.1782          | 0.9457   | 0.8632    | 0.8368 | 0.8484 |
 
 
 ### Framework versions