edwin923
/

FYP2022

Text Classification

generated_from_keras_callback

text-embeddings-inference

Model card Files Files and versions

edwin923 commited on May 17, 2022

Commit

7d804b8

·

1 Parent(s): 074a319

End of training

Files changed (2) hide show

README.md +10 -5
tf_model.h5 +1 -1

README.md CHANGED Viewed

@@ -14,10 +14,10 @@ probably proofread and complete it, then remove this comment. -->
 This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Train Loss: 0.9481
-- Train Sparse Categorical Accuracy: 0.5960
-- Train Sparse Top 3 Categorical Accuracy: 0.9512
-- Epoch: 0
 ## Model description
@@ -43,7 +43,12 @@ The following hyperparameters were used during training:
 | Train Loss | Train Sparse Categorical Accuracy | Train Sparse Top 3 Categorical Accuracy | Epoch |
 |:----------:|:---------------------------------:|:---------------------------------------:|:-----:|
-| 0.9481     | 0.5960                            | 0.9512                                  | 0     |
 ### Framework versions

 This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on an unknown dataset.
 It achieves the following results on the evaluation set:
+- Train Loss: 0.6016
+- Train Sparse Categorical Accuracy: 0.7503
+- Train Sparse Top 3 Categorical Accuracy: 0.9901
+- Epoch: 5
 ## Model description
 | Train Loss | Train Sparse Categorical Accuracy | Train Sparse Top 3 Categorical Accuracy | Epoch |
 |:----------:|:---------------------------------:|:---------------------------------------:|:-----:|
+| 0.9433     | 0.5975                            | 0.9523                                  | 0     |
+| 0.8257     | 0.6498                            | 0.9704                                  | 1     |
+| 0.7625     | 0.6765                            | 0.9778                                  | 2     |
+| 0.7062     | 0.7014                            | 0.9832                                  | 3     |
+| 0.6526     | 0.7263                            | 0.9872                                  | 4     |
+| 0.6016     | 0.7503                            | 0.9901                                  | 5     |
 ### Framework versions

tf_model.h5 CHANGED Viewed

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

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