tkuye
/

binary-skills-classifier

Text Classification

Generated from Trainer

text-embeddings-inference

Model card Files Files and versions

Tosin Kuye commited on Sep 17, 2022

Commit

7143bed

·

1 Parent(s): 55ddf9f

update model card README.md

Files changed (1) hide show

README.md +6 -6

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](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.1304
-- Accuracy: 0.9727
 ## Model description
@@ -48,10 +48,10 @@ The following hyperparameters were used during training:
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 0.09          | 1.0   | 1557 | 0.0823          | 0.9723   |
-| 0.0571        | 2.0   | 3114 | 0.0934          | 0.9741   |
-| 0.0321        | 3.0   | 4671 | 0.1078          | 0.9736   |
-| 0.0213        | 4.0   | 6228 | 0.1304          | 0.9727   |
 ### Framework versions

 This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
 It achieves the following results on the evaluation set:
+- Loss: 0.1373
+- Accuracy: 0.9702
 ## Model description
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
+| 0.098         | 1.0   | 1557 | 0.0917          | 0.9663   |
+| 0.0678        | 2.0   | 3114 | 0.0982          | 0.9712   |
+| 0.0344        | 3.0   | 4671 | 0.1140          | 0.9712   |
+| 0.0239        | 4.0   | 6228 | 0.1373          | 0.9702   |
 ### Framework versions