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