ViditRaj
/

BERT_Ads_Classifier

Text Classification

generated_from_keras_callback

text-embeddings-inference

Model card Files Files and versions

ViditRaj commited on Mar 4, 2023

Commit

265043b

·

1 Parent(s): 750284b

Training in progress epoch 3

Files changed (2) hide show

README.md +6 -3
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 [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.0567
-- Validation Loss: 0.2058
 - Train Accuracy: 0.9348
-- Epoch: 0
 ## Model description
@@ -44,6 +44,9 @@ The following hyperparameters were used during training:
 | Train Loss | Validation Loss | Train Accuracy | Epoch |
 |:----------:|:---------------:|:--------------:|:-----:|
 | 0.0567     | 0.2058          | 0.9348         | 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: 0.0373
+- Validation Loss: 0.2213
 - Train Accuracy: 0.9348
+- Epoch: 3
 ## Model description
 | Train Loss | Validation Loss | Train Accuracy | Epoch |
 |:----------:|:---------------:|:--------------:|:-----:|
 | 0.0567     | 0.2058          | 0.9348         | 0     |
+| 0.0407     | 0.2213          | 0.9348         | 1     |
+| 0.0373     | 0.2213          | 0.9348         | 2     |
+| 0.0373     | 0.2213          | 0.9348         | 3     |
 ### Framework versions

tf_model.h5 CHANGED Viewed

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

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