Commit
•
5cf2c0a
1
Parent(s):
6c54008
Upload TFDistilBertForSequenceClassification
Browse files- README.md +8 -10
- tf_model.h5 +1 -1
README.md
CHANGED
@@ -14,11 +14,11 @@ probably proofread and complete it, then remove this comment. -->
|
|
14 |
|
15 |
This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
|
16 |
It achieves the following results on the evaluation set:
|
17 |
-
- Train Loss: 0.
|
18 |
-
- Train Sparse Categorical Accuracy: 0.
|
19 |
-
- Validation Loss: 0.
|
20 |
-
- Validation Sparse Categorical Accuracy: 0.
|
21 |
-
- Epoch:
|
22 |
|
23 |
## Model description
|
24 |
|
@@ -44,11 +44,9 @@ The following hyperparameters were used during training:
|
|
44 |
|
45 |
| Train Loss | Train Sparse Categorical Accuracy | Validation Loss | Validation Sparse Categorical Accuracy | Epoch |
|
46 |
|:----------:|:---------------------------------:|:---------------:|:--------------------------------------:|:-----:|
|
47 |
-
| 0.
|
48 |
-
| 0.
|
49 |
-
| 0.
|
50 |
-
| 0.0437 | 0.9865 | 0.2782 | 0.9240 | 3 |
|
51 |
-
| 0.0254 | 0.9922 | 0.3001 | 0.9276 | 4 |
|
52 |
|
53 |
|
54 |
### Framework versions
|
|
|
14 |
|
15 |
This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
|
16 |
It achieves the following results on the evaluation set:
|
17 |
+
- Train Loss: 0.0859
|
18 |
+
- Train Sparse Categorical Accuracy: 0.9708
|
19 |
+
- Validation Loss: 0.2284
|
20 |
+
- Validation Sparse Categorical Accuracy: 0.9244
|
21 |
+
- Epoch: 2
|
22 |
|
23 |
## Model description
|
24 |
|
|
|
44 |
|
45 |
| Train Loss | Train Sparse Categorical Accuracy | Validation Loss | Validation Sparse Categorical Accuracy | Epoch |
|
46 |
|:----------:|:---------------------------------:|:---------------:|:--------------------------------------:|:-----:|
|
47 |
+
| 0.2731 | 0.8882 | 0.1975 | 0.9200 | 0 |
|
48 |
+
| 0.1554 | 0.9437 | 0.1929 | 0.9229 | 1 |
|
49 |
+
| 0.0859 | 0.9708 | 0.2284 | 0.9244 | 2 |
|
|
|
|
|
50 |
|
51 |
|
52 |
### Framework versions
|
tf_model.h5
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 267955144
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ca9aa14c6b15fc4e722c47c71d5f68742d84c04b75bfd6ea1cf34f82afc9d95e
|
3 |
size 267955144
|