gualti
/

imdb_model

@@ -16,9 +16,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: 0.1071
-- Validation Loss: 0.0883
-- Train Accuracy: 0.968
 - Epoch: 2
 ## Model description
@@ -45,9 +45,9 @@ The following hyperparameters were used during training:
 | Train Loss | Validation Loss | Train Accuracy | Epoch |
 |:----------:|:---------------:|:--------------:|:-----:|
-| 0.4326     | 0.2084          | 0.924          | 0     |
-| 0.1850     | 0.1201          | 0.962          | 1     |
-| 0.1071     | 0.0883          | 0.968          | 2     |
 ### 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: 0.1219
+- Validation Loss: 0.0976
+- Train Accuracy: 0.976
 - Epoch: 2
 ## Model description
 | Train Loss | Validation Loss | Train Accuracy | Epoch |
 |:----------:|:---------------:|:--------------:|:-----:|
+| 0.4391     | 0.1931          | 0.93           | 0     |
+| 0.1996     | 0.1249          | 0.96           | 1     |
+| 0.1219     | 0.0976          | 0.976          | 2     |
 ### Framework versions

tf_model.h5 CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:389c83e3dad77ff28a84622f546d9f3fadf3eb6ec5e41581644a4745aa2342a3
 size 267951808