End of training

README.md

@@ -1,6 +1,6 @@
 ---
-license: apache-2.0
-base_model: distilbert-base-uncased
+license: mit
+base_model: nlptown/bert-base-multilingual-uncased-sentiment
 tags:
 - generated_from_trainer
 datasets:
@@ -23,10 +23,10 @@ model-index:
     metrics:
     - name: Accuracy
       type: accuracy
-      value: 0.7408
+      value: 0.7576
     - name: F1
       type: f1
-      value: 0.6839666291008939
+      value: 0.7044852616595344
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -34,11 +34,11 @@ should probably proofread and complete it, then remove this comment. -->
 
 # amazon-reviews-finetuning-distilbert-base-uncased
 
-This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the amazon_reviews_multi dataset.
+This model is a fine-tuned version of [nlptown/bert-base-multilingual-uncased-sentiment](https://huggingface.co/nlptown/bert-base-multilingual-uncased-sentiment) on the amazon_reviews_multi dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.6228
-- Accuracy: 0.7408
-- F1: 0.6840
+- Loss: 0.6020
+- Accuracy: 0.7576
+- F1: 0.7045
 
 ## Model description
 
@@ -69,8 +69,8 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1     |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
-| 0.6992        | 1.0   | 625  | 0.6202          | 0.74     | 0.6720 |
-| 0.5651        | 2.0   | 1250 | 0.6228          | 0.7408   | 0.6840 |
+| 0.6077        | 1.0   | 625  | 0.5697          | 0.7588   | 0.7045 |
+| 0.5097        | 2.0   | 1250 | 0.6020          | 0.7576   | 0.7045 |
 
 
 ### Framework versions

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a78962913e83e19236c3f6ee40ec2aec68d99d16257c51d3ec8a7311f0f7dbc3
+oid sha256:4bd74d937edfad3e8b567709a8caf1cc8701d3cb256e4fbc674461e02e9ca17c
 size 669503281