tlapusan
/

distilbert-base-uncased-finetuned-imdb

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

tlapusan commited on Feb 17, 2023

Commit

2ad15b4

•

1 Parent(s): 389c478

update model card README.md

Files changed (1) hide show

README.md +4 -4

README.md CHANGED Viewed

@@ -14,7 +14,7 @@ should probably proofread and complete it, then remove this comment. -->
 This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 2.1639
 ## Model description
@@ -46,9 +46,9 @@ The following hyperparameters were used during training:
 | Training Loss | Epoch | Step | Validation Loss |
 |:-------------:|:-----:|:----:|:---------------:|
-| 2.7695        | 1.0   | 90   | 2.3614          |
-| 2.3627        | 2.0   | 180  | 2.1959          |
-| 2.227         | 3.0   | 270  | 2.1313          |
 ### Framework versions

 This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the None dataset.
 It achieves the following results on the evaluation set:
+- Loss: 2.1730
 ## Model description
 | Training Loss | Epoch | Step | Validation Loss |
 |:-------------:|:-----:|:----:|:---------------:|
+| 2.7736        | 1.0   | 97   | 2.3876          |
+| 2.3667        | 2.0   | 194  | 2.1674          |
+| 2.2575        | 3.0   | 291  | 2.1479          |
 ### Framework versions