anvayS
/

reddit-aita-classifier

Text Classification

Generated from Trainer

text-embeddings-inference

Model card Files Files and versions

Metrics Training metrics Community

anvayS commited on Jun 13, 2022

Commit

6cb9a1b

·

1 Parent(s): e77ed89

update model card README.md

Files changed (1) hide show

README.md +8 -8

README.md CHANGED Viewed

@@ -16,8 +16,8 @@ should probably proofread and complete it, then remove this comment. -->
 This model is a fine-tuned version of [distilbert-base-uncased-finetuned-sst-2-english](https://huggingface.co/distilbert-base-uncased-finetuned-sst-2-english) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.5896
-- Accuracy: 0.7247
 ## Model description
@@ -46,12 +46,12 @@ The following hyperparameters were used during training:
 ### Training results
-| Training Loss | Epoch | Step  | Validation Loss | Accuracy |
-|:-------------:|:-----:|:-----:|:---------------:|:--------:|
-| 0.6063        | 1.0   | 2500  | 0.6102          | 0.7247   |
-| 0.5872        | 2.0   | 5000  | 0.5918          | 0.7247   |
-| 0.5861        | 3.0   | 7500  | 0.5908          | 0.7247   |
-| 0.5878        | 4.0   | 10000 | 0.5896          | 0.7247   |
 ### Framework versions

 This model is a fine-tuned version of [distilbert-base-uncased-finetuned-sst-2-english](https://huggingface.co/distilbert-base-uncased-finetuned-sst-2-english) on an unknown dataset.
 It achieves the following results on the evaluation set:
+- Loss: 0.1667
+- Accuracy: 0.9497
 ## Model description
 ### Training results
+| Training Loss | Epoch | Step | Validation Loss | Accuracy |
+|:-------------:|:-----:|:----:|:---------------:|:--------:|
+| 0.5866        | 1.0   | 1250 | 0.5692          | 0.7247   |
+| 0.5638        | 2.0   | 2500 | 0.4841          | 0.7813   |
+| 0.4652        | 3.0   | 3750 | 0.2712          | 0.9077   |
+| 0.3088        | 4.0   | 5000 | 0.1667          | 0.9497   |
 ### Framework versions