andywedlake
/

test-finetuned-imdb

generated_from_keras_callback

Inference Endpoints

Model card Files Files and versions Community

andywedlake commited on May 26, 2022

Commit

7a658e0

•

1 Parent(s): 8080098

End of training

Files changed (2) hide show

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

README.md CHANGED Viewed

@@ -14,9 +14,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: 2.6663
-- Validation Loss: 2.5238
-- Epoch: 0
 ## Model description
@@ -43,6 +43,15 @@ The following hyperparameters were used during training:
 | Train Loss | Validation Loss | Epoch |
 |:----------:|:---------------:|:-----:|
 | 2.6663     | 2.5238          | 0     |
 ### 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: 1.7505
+- Validation Loss: 2.0472
+- Epoch: 9
 ## Model description
 | Train Loss | Validation Loss | Epoch |
 |:----------:|:---------------:|:-----:|
 | 2.6663     | 2.5238          | 0     |
+| 2.4783     | 2.6058          | 1     |
+| 2.4284     | 2.5982          | 2     |
+| 2.3804     | 2.5057          | 3     |
+| 2.3487     | 2.6968          | 4     |
+| 2.1253     | 2.1361          | 5     |
+| 2.0700     | 2.2953          | 6     |
+| 1.9491     | 2.3122          | 7     |
+| 1.7558     | 2.5881          | 8     |
+| 1.7505     | 2.0472          | 9     |
 ### Framework versions

tf_model.h5 CHANGED Viewed

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

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