End of training

README.md

@@ -1,6 +1,6 @@
 ---
-license: mit
-base_model: roberta-base
+license: apache-2.0
+base_model: distilbert-base-uncased
 tags:
 - generated_from_trainer
 model-index:
@@ -13,10 +13,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # Sentiment-Analysis-Model
 
-This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on an unknown dataset.
+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.6488
-- F1 Score: 0.7281
+- Loss: 0.6064
+- F1 Score: 0.7343
 
 ## Model description
 
@@ -47,16 +47,16 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | F1 Score |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 0.8011        | 0.5   | 500  | 0.8189          | 0.6563   |
-| 0.7437        | 1.0   | 1000 | 0.7877          | 0.6793   |
-| 0.6996        | 1.5   | 1500 | 0.6979          | 0.6996   |
-| 0.6886        | 2.0   | 2000 | 0.6893          | 0.6910   |
-| 0.6384        | 2.5   | 2500 | 0.8285          | 0.7133   |
-| 0.6254        | 3.0   | 3000 | 0.6488          | 0.7281   |
-| 0.56          | 3.5   | 3500 | 0.6755          | 0.7402   |
-| 0.5731        | 4.0   | 4000 | 0.6775          | 0.7456   |
-| 0.504         | 4.5   | 4500 | 0.7232          | 0.7407   |
-| 0.4839        | 5.0   | 5000 | 0.7219          | 0.7512   |
+| 0.7496        | 0.5   | 500  | 0.7056          | 0.6853   |
+| 0.6707        | 1.0   | 1000 | 0.6539          | 0.7240   |
+| 0.5505        | 1.5   | 1500 | 0.6562          | 0.7459   |
+| 0.5418        | 2.0   | 2000 | 0.6064          | 0.7343   |
+| 0.3717        | 2.5   | 2500 | 0.9214          | 0.7443   |
+| 0.3713        | 3.0   | 3000 | 0.8631          | 0.7397   |
+| 0.2445        | 3.5   | 3500 | 1.0991          | 0.7386   |
+| 0.2438        | 4.0   | 4000 | 1.1678          | 0.7402   |
+| 0.1594        | 4.5   | 4500 | 1.3197          | 0.7314   |
+| 0.1554        | 5.0   | 5000 | 1.3487          | 0.7343   |
 
 
 ### Framework versions

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:439343f1b490721664ce06de3d77b966a8ddd9301473069aace7aa58235448c0
+oid sha256:60ca994c9ea954579ebf23b1bf37dd04bd769ce6bd845d5f3606b4447c771750
 size 267858605