Dhmo1900
/

distilbert-base-uncased-finetuned-imdb

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Community

Dhmo1900 commited on Nov 17, 2024

Commit

f8d5176

·

verified ·

1 Parent(s): 88f4ac3

Model save

Files changed (1) hide show

README.md +4 -4

README.md CHANGED Viewed

@@ -16,7 +16,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 an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 2.4970
 ## Model description
@@ -48,9 +48,9 @@ The following hyperparameters were used during training:
 | Training Loss | Epoch | Step | Validation Loss |
 |:-------------:|:-----:|:----:|:---------------:|
-| No log        | 1.0   | 40   | 2.5363          |
-| No log        | 2.0   | 80   | 2.4995          |
-| No log        | 3.0   | 120  | 2.5049          |
 ### Framework versions

 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: 2.4598
 ## Model description
 | Training Loss | Epoch | Step | Validation Loss |
 |:-------------:|:-----:|:----:|:---------------:|
+| No log        | 1.0   | 79   | 2.5011          |
+| No log        | 2.0   | 158  | 2.4250          |
+| No log        | 3.0   | 237  | 2.4225          |
 ### Framework versions