RathindraH
/

News-headline-classification

Text Classification

generated_from_keras_callback

Model card Files Files and versions

RathindraH commited on Jul 27

Commit

b87ca9d

·

1 Parent(s): 741aa76

Training in progress epoch 4

Files changed (2) hide show

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

README.md CHANGED Viewed

@@ -15,11 +15,11 @@ probably proofread and complete it, then remove this comment. -->
 This model is a fine-tuned version of [ProsusAI/finbert](https://huggingface.co/ProsusAI/finbert) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Train Loss: 0.0054
-- Validation Loss: 0.8142
-- Train Accuracy: 0.8690
-- Train F1 Macro: 0.8569
-- Epoch: 3
 ## Model description
@@ -49,6 +49,7 @@ The following hyperparameters were used during training:
 | 0.0050     | 0.7373          | 0.8726         | 0.8563         | 1     |
 | 0.0023     | 0.7720          | 0.8738         | 0.8586         | 2     |
 | 0.0054     | 0.8142          | 0.8690         | 0.8569         | 3     |
 ### Framework versions

 This model is a fine-tuned version of [ProsusAI/finbert](https://huggingface.co/ProsusAI/finbert) on an unknown dataset.
 It achieves the following results on the evaluation set:
+- Train Loss: 0.0039
+- Validation Loss: 0.7602
+- Train Accuracy: 0.8679
+- Train F1 Macro: 0.8537
+- Epoch: 4
 ## Model description
 | 0.0050     | 0.7373          | 0.8726         | 0.8563         | 1     |
 | 0.0023     | 0.7720          | 0.8738         | 0.8586         | 2     |
 | 0.0054     | 0.8142          | 0.8690         | 0.8569         | 3     |
+| 0.0039     | 0.7602          | 0.8679         | 0.8537         | 4     |
 ### Framework versions

tf_model.h5 CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:4ed58b87e8ec340eb55af90328ed24add1f484b6e97a3f940c43df2abd6ff4b1
 size 438226204