an-imdb-classifier

README.md

@@ -4,19 +4,22 @@ license: apache-2.0
 base_model: distilbert-base-uncased
 tags:
 - generated_from_trainer
+metrics:
+- accuracy
 model-index:
-- name: results
+- name: an-imdb-classifier
   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. -->
 
-# results
+# an-imdb-classifier
 
 This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0000
+- Loss: 0.3635
+- Accuracy: 0.898
 
 ## Model description
 
@@ -45,11 +48,11 @@ The following hyperparameters were used during training:
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss |
-|:-------------:|:-----:|:----:|:---------------:|
-| No log        | 1.0   | 310  | 0.0002          |
-| 0.0103        | 2.0   | 620  | 0.0001          |
-| 0.0103        | 3.0   | 930  | 0.0000          |
+| Training Loss | Epoch | Step | Validation Loss | Accuracy |
+|:-------------:|:-----:|:----:|:---------------:|:--------:|
+| No log        | 1.0   | 313  | 0.3199          | 0.866    |
+| 0.2966        | 2.0   | 626  | 0.3023          | 0.89     |
+| 0.2966        | 3.0   | 939  | 0.3635          | 0.898    |
 
 
 ### Framework versions