andywedlake
/

test-finetuned-imdb

generated_from_keras_callback

Inference Endpoints

Model card Files Files and versions Community

andywedlake commited on Jul 9

Commit

5352a58

•

1 Parent(s): 7740fe0

Training in progress epoch 5

Files changed (2) hide show

README.md +4 -3
tf_model.h5 +1 -1

README.md CHANGED Viewed

@@ -15,9 +15,9 @@ 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:
-- Train Loss: 3.7038
-- Validation Loss: 4.0675
-- Epoch: 4
 ## Model description
@@ -48,6 +48,7 @@ The following hyperparameters were used during training:
 | 4.4752     | 4.8518          | 2     |
 | 4.4115     | 2.9955          | 3     |
 | 3.7038     | 4.0675          | 4     |
 ### 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:
+- Train Loss: 3.7644
+- Validation Loss: 2.6604
+- Epoch: 5
 ## Model description
 | 4.4752     | 4.8518          | 2     |
 | 4.4115     | 2.9955          | 3     |
 | 3.7038     | 4.0675          | 4     |
+| 3.7644     | 2.6604          | 5     |
 ### Framework versions

tf_model.h5 CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f2c226a36fd707252f5ba1108b364be151979b66f63ddef91d09fa9f650b33c5
 size 363423648

 version https://git-lfs.github.com/spec/v1
+oid sha256:c52e6a0250716a4dc3b9f63ee693bdcbbdb7e600d12f3749809a2b2f4059b37a
 size 363423648