Mou11209203
/

distilbert-base-uncased_stock_classification_finetuned_news_all_epoch2

Text Classification

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Community

Mou11209203 commited on Apr 12

Commit

88a28ba

•

1 Parent(s): b485a15

End of training

Files changed (1) hide show

README.md +4 -4

README.md CHANGED Viewed

@@ -17,8 +17,8 @@ should probably proofread and complete it, then remove this comment. -->
 This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.3271
-- Accuracy: 0.8324
 ## Model description
@@ -49,8 +49,8 @@ The following hyperparameters were used during training:
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 0.3658        | 1.0   | 2384 | 0.3360          | 0.8184   |
-| 0.3195        | 2.0   | 4768 | 0.3271          | 0.8324   |
 ### Framework versions

 This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on an unknown dataset.
 It achieves the following results on the evaluation set:
+- Loss: 0.3245
+- Accuracy: 0.8301
 ## Model description
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
+| 0.3687        | 1.0   | 2384 | 0.3379          | 0.8178   |
+| 0.3242        | 2.0   | 4768 | 0.3245          | 0.8301   |
 ### Framework versions