Initial model upload

README.md

@@ -1,27 +1,27 @@
 ---
 library_name: transformers
 license: apache-2.0
-base_model: bert-base-multilingual-uncased
+base_model: bert-base-uncased
 tags:
 - generated_from_trainer
 metrics:
 - accuracy
 - f1
 model-index:
-- name: mbert-sst2
+- name: bert-sst2
   results: []
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 should probably proofread and complete it, then remove this comment. -->
 
-# mbert-sst2
+# bert-sst2
 
-This model is a fine-tuned version of [bert-base-multilingual-uncased](https://huggingface.co/bert-base-multilingual-uncased) on an unknown dataset.
+This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.3137
-- Accuracy: 0.8968
-- F1: 0.8968
+- Loss: 0.2521
+- Accuracy: 0.9174
+- F1: 0.9174
 
 ## Model description
 
@@ -53,7 +53,7 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1     |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
-| 0.2213        | 1.0   | 4210 | 0.3137          | 0.8968   | 0.8968 |
+| 0.1712        | 1.0   | 4210 | 0.2521          | 0.9174   | 0.9174 |
 
 
 ### Framework versions