Iiro/bert_reviews

README.md

@@ -1,28 +1,13 @@
 ---
 license: apache-2.0
-base_model: distilbert-base-uncased
+base_model: bert-base-cased
 tags:
 - generated_from_trainer
 datasets:
 - amazon_reviews_multi
-metrics:
-- accuracy
 model-index:
 - name: bert_reviews
- results:
- - task:
- name: Text Classification
- type: text-classification
- dataset:
- name: amazon_reviews_multi
- type: amazon_reviews_multi
- config: en
- split: test
- args: en
- metrics:
- - name: Accuracy
- type: accuracy
- value: 0.6062
+ results: []
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -30,10 +15,15 @@ should probably proofread and complete it, then remove this comment. -->
 
 # bert_reviews
 
-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 [bert-base-cased](https://huggingface.co/bert-base-cased) on the amazon_reviews_multi dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.9204
-- Accuracy: 0.6062
+- eval_loss: 0.8063
+- eval_accuracy: 0.647
+- eval_runtime: 38.7598
+- eval_samples_per_second: 129.0
+- eval_steps_per_second: 16.125
+- epoch: 0.52
+- step: 13000
 
 ## Model description
 
@@ -60,32 +50,6 @@ The following hyperparameters were used during training:
 - lr_scheduler_type: linear
 - training_steps: 20000
 
-### Training results
-
-| Training Loss | Epoch | Step | Validation Loss | Accuracy |
-|:-------------:|:-----:|:-----:|:---------------:|:--------:|
-| 0.8812 | 0.04 | 1000 | 0.9970 | 0.5738 |
-| 0.8495 | 0.08 | 2000 | 1.0120 | 0.569 |
-| 0.8067 | 0.12 | 3000 | 1.0442 | 0.5766 |
-| 0.7934 | 0.16 | 4000 | 1.0629 | 0.5772 |
-| 0.7845 | 0.2 | 5000 | 1.0236 | 0.5876 |
-| 0.9033 | 0.24 | 6000 | 0.9822 | 0.5774 |
-| 0.8993 | 0.28 | 7000 | 0.9693 | 0.5816 |
-| 0.9012 | 0.32 | 8000 | 1.0075 | 0.5738 |
-| 0.873 | 0.36 | 9000 | 0.9663 | 0.5886 |
-| 0.9376 | 0.4 | 10000 | 0.9447 | 0.5816 |
-| 0.9398 | 0.44 | 11000 | 0.9509 | 0.5802 |
-| 0.9402 | 0.48 | 12000 | 0.9561 | 0.5916 |
-| 0.9247 | 0.52 | 13000 | 0.9303 | 0.6008 |
-| 0.9247 | 0.56 | 14000 | 0.9241 | 0.5998 |
-| 0.9192 | 0.6 | 15000 | 0.9276 | 0.6104 |
-| 0.907 | 0.64 | 16000 | 0.9251 | 0.603 |
-| 0.9177 | 0.68 | 17000 | 0.9198 | 0.6056 |
-| 0.9129 | 0.72 | 18000 | 0.9167 | 0.6078 |
-| 0.8948 | 0.76 | 19000 | 0.9213 | 0.604 |
-| 0.906 | 0.8 | 20000 | 0.9204 | 0.6062 |
-
-
 ### Framework versions
 
 - Transformers 4.34.1

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:20ce5fc14cf92e9f8930a90fb2ea322cc504099e9ba1b5efbe90696a99082aaf
+oid sha256:a3ea5ad386bdff62d86aa6b74df64072f65d98cb035d90de3736bef025fcdd51
 size 433325294