RathindraH
/

News-headline-classification

Text Classification

generated_from_keras_callback

Model card Files Files and versions

RathindraH commited on Jul 27

Commit

741aa76

·

1 Parent(s): e462def

Training in progress epoch 3

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.0023
-- Validation Loss: 0.7720
-- Train Accuracy: 0.8738
-- Train F1 Macro: 0.8586
-- Epoch: 2
 ## Model description
@@ -48,6 +48,7 @@ The following hyperparameters were used during training:
 | 0.0085     | 0.7174          | 0.8762         | 0.8642         | 0     |
 | 0.0050     | 0.7373          | 0.8726         | 0.8563         | 1     |
 | 0.0023     | 0.7720          | 0.8738         | 0.8586         | 2     |
 ### 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.0054
+- Validation Loss: 0.8142
+- Train Accuracy: 0.8690
+- Train F1 Macro: 0.8569
+- Epoch: 3
 ## Model description
 | 0.0085     | 0.7174          | 0.8762         | 0.8642         | 0     |
 | 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

tf_model.h5 CHANGED Viewed

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

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