TeamResearch
/

sentiment-model-saagie

Text Classification

Generated from Trainer

text-embeddings-inference

Model card Files Files and versions

TeamResearch commited on Dec 19, 2023

Commit

89a7918

·

1 Parent(s): edba215

update model card README.md

Files changed (1) hide show

README.md +4 -4

README.md CHANGED Viewed

@@ -29,7 +29,7 @@ should probably proofread and complete it, then remove this comment. -->
 This model is a fine-tuned version of [prajjwal1/bert-tiny](https://huggingface.co/prajjwal1/bert-tiny) on the sst2 dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.5436
 - Accuracy: 0.7833
 ## Model description
@@ -61,9 +61,9 @@ The following hyperparameters were used during training:
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 0.5215        | 1.0   | 1500 | 0.4752          | 0.775    |
-| 0.3796        | 2.0   | 3000 | 0.5361          | 0.7683   |
-| 0.3346        | 3.0   | 4500 | 0.5436          | 0.7833   |
 ### Framework versions

 This model is a fine-tuned version of [prajjwal1/bert-tiny](https://huggingface.co/prajjwal1/bert-tiny) on the sst2 dataset.
 It achieves the following results on the evaluation set:
+- Loss: 0.5443
 - Accuracy: 0.7833
 ## Model description
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
+| 0.5214        | 1.0   | 1500 | 0.4842          | 0.775    |
+| 0.3834        | 2.0   | 3000 | 0.5233          | 0.7767   |
+| 0.3417        | 3.0   | 4500 | 0.5443          | 0.7833   |
 ### Framework versions