filippo571
/

pull_request_comments_model

Text Classification

generated_from_keras_callback

text-embeddings-inference

Model card Files Files and versions

filippo571 commited on Mar 7, 2023

Commit

1b9e368

·

1 Parent(s): da71f2e

Training in progress epoch 2

Files changed (2) hide show

README.md +6 -4
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: 1.2707
-- Validation Loss: 1.0862
-- Train Accuracy: 0.5812
-- Epoch: 0
 ## Model description
@@ -44,6 +44,8 @@ The following hyperparameters were used during training:
 | Train Loss | Validation Loss | Train Accuracy | Epoch |
 |:----------:|:---------------:|:--------------:|:-----:|
 | 1.2707     | 1.0862          | 0.5812         | 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.8048
+- Validation Loss: 0.7373
+- Train Accuracy: 0.7265
+- Epoch: 2
 ## Model description
 | Train Loss | Validation Loss | Train Accuracy | Epoch |
 |:----------:|:---------------:|:--------------:|:-----:|
 | 1.2707     | 1.0862          | 0.5812         | 0     |
+| 1.0411     | 0.8534          | 0.7179         | 1     |
+| 0.8048     | 0.7373          | 0.7265         | 2     |
 ### Framework versions

tf_model.h5 CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:4910025126f06c42c750d2537a8e3789ef6dce548086a56a21084cec0ebb55b7
 size 267957952