End of training

README.md

@@ -1,6 +1,5 @@
 ---
-license: apache-2.0
-base_model: distilbert-base-uncased
+base_model: finiteautomata/bertweet-base-sentiment-analysis
 tags:
 - generated_from_trainer
 metrics:
@@ -15,10 +14,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # my_awesome_model_IMDB
 
-This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
+This model is a fine-tuned version of [finiteautomata/bertweet-base-sentiment-analysis](https://huggingface.co/finiteautomata/bertweet-base-sentiment-analysis) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.4229
-- Accuracy: 0.9307
+- Loss: 0.6664
+- Accuracy: 0.8949
 
 ## Model description
 
@@ -49,16 +48,16 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 0.2892        | 1.0   | 782  | 0.2075          | 0.9192   |
-| 0.151         | 2.0   | 1564 | 0.2034          | 0.9294   |
-| 0.103         | 3.0   | 2346 | 0.2369          | 0.9270   |
-| 0.0567        | 4.0   | 3128 | 0.2920          | 0.9306   |
-| 0.0415        | 5.0   | 3910 | 0.3699          | 0.9275   |
-| 0.025         | 6.0   | 4692 | 0.3560          | 0.9300   |
-| 0.0204        | 7.0   | 5474 | 0.3690          | 0.9308   |
-| 0.0125        | 8.0   | 6256 | 0.4119          | 0.9300   |
-| 0.0117        | 9.0   | 7038 | 0.4176          | 0.9310   |
-| 0.0065        | 10.0  | 7820 | 0.4229          | 0.9307   |
+| 0.3261        | 1.0   | 782  | 0.2674          | 0.8903   |
+| 0.2072        | 2.0   | 1564 | 0.3035          | 0.8820   |
+| 0.1408        | 3.0   | 2346 | 0.3532          | 0.8967   |
+| 0.0876        | 4.0   | 3128 | 0.4793          | 0.8922   |
+| 0.0661        | 5.0   | 3910 | 0.4755          | 0.8925   |
+| 0.0373        | 6.0   | 4692 | 0.5159          | 0.8937   |
+| 0.034         | 7.0   | 5474 | 0.5527          | 0.8923   |
+| 0.0264        | 8.0   | 6256 | 0.6391          | 0.8947   |
+| 0.0179        | 9.0   | 7038 | 0.6491          | 0.8942   |
+| 0.0094        | 10.0  | 7820 | 0.6664          | 0.8949   |
 
 
 ### Framework versions

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:06a3848d845b1f0613e0d478e46988ba34c7dda92a0d2fbe2969f3639028ee4e
+oid sha256:2bd8fcc972c53d32c69c7fe64e3ba0b9c0d5539252afed54ba4ee3848a71f884
 size 539678065