slickdata
/

finetuned-Sentiment-classfication-DistilBert-model

Text Classification

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

slickdata commited on Jul 23, 2023

Commit

ddc8c9b

·

1 Parent(s): 2fc35fe

update model card README.md

Files changed (1) hide show

README.md +7 -7

README.md CHANGED Viewed

@@ -15,8 +15,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.2764
-- Rmse: 0.2925
 ## Model description
@@ -50,11 +50,11 @@ The following hyperparameters were used during training:
 | Training Loss | Epoch | Step | Validation Loss | Rmse   |
 |:-------------:|:-----:|:----:|:---------------:|:------:|
-| 0.7292        | 2.72  | 500  | 0.3805          | 0.5119 |
-| 0.1778        | 5.43  | 1000 | 0.2802          | 0.3530 |
-| 0.0487        | 8.15  | 1500 | 0.2764          | 0.2925 |
-| 0.0209        | 10.86 | 2000 | 0.2921          | 0.2860 |
-| 0.0113        | 13.58 | 2500 | 0.3244          | 0.2884 |
 ### 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.2934
+- Rmse: 0.3657
 ## Model description
 | Training Loss | Epoch | Step | Validation Loss | Rmse   |
 |:-------------:|:-----:|:----:|:---------------:|:------:|
+| 0.7234        | 2.72  | 500  | 0.4388          | 0.5622 |
+| 0.1723        | 5.43  | 1000 | 0.2934          | 0.3657 |
+| 0.0454        | 8.15  | 1500 | 0.3104          | 0.3099 |
+| 0.021         | 10.86 | 2000 | 0.3128          | 0.2884 |
+| 0.013         | 13.58 | 2500 | 0.3428          | 0.2907 |
 ### Framework versions