felixshier
/

csc-01-bert-finetuned

Text Classification

generated_from_keras_callback

Inference Endpoints

Model card Files Files and versions Community

felixshier commited on Aug 9, 2023

Commit

f6c412c

•

1 Parent(s): 4732857

Upload TFBertForSequenceClassification

Files changed (2) hide show

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

README.md CHANGED Viewed

@@ -15,10 +15,10 @@ probably proofread and complete it, then remove this comment. -->
 This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Train Loss: 0.6165
-- Validation Loss: 0.6875
-- Train Precision: 0.4918
-- Epoch: 3
 ## Model description
@@ -44,10 +44,12 @@ The following hyperparameters were used during training:
 | Train Loss | Validation Loss | Train Precision | Epoch |
 |:----------:|:---------------:|:---------------:|:-----:|
-| 0.7154     | 0.6899          | 0.4861          | 0     |
-| 0.6647     | 0.6831          | 0.5088          | 1     |
-| 0.6534     | 0.6986          | 0.4795          | 2     |
-| 0.6165     | 0.6875          | 0.4918          | 3     |
 ### Framework versions

 This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on an unknown dataset.
 It achieves the following results on the evaluation set:
+- Train Loss: 0.4789
+- Validation Loss: 0.7231
+- Train Precision: 0.6429
+- Epoch: 5
 ## Model description
 | Train Loss | Validation Loss | Train Precision | Epoch |
 |:----------:|:---------------:|:---------------:|:-----:|
+| 0.7100     | 0.7421          | 0.0             | 0     |
+| 0.6764     | 0.6861          | 0.625           | 1     |
+| 0.6311     | 0.6838          | 0.5862          | 2     |
+| 0.5909     | 0.7072          | 0.6286          | 3     |
+| 0.5413     | 0.7504          | 0.6667          | 4     |
+| 0.4789     | 0.7231          | 0.6429          | 5     |
 ### Framework versions

tf_model.h5 CHANGED Viewed

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

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