ShreyaR
/

finetuned-distil-bert-depression

Text Classification

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

ShreyaR commited on May 3, 2022

Commit

c0eba01

•

1 Parent(s): 2d4abb1

Update README.md

Files changed (1) hide show

README.md +5 -5

README.md CHANGED Viewed

@@ -16,8 +16,8 @@ should 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:
-- Loss: 0.1495
-- Accuracy: 0.9645
 ## Model description
@@ -48,9 +48,9 @@ The following hyperparameters were used during training:
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| No log        | 1.0   | 38   | 0.2303          | 0.9205   |
-| No log        | 2.0   | 76   | 0.1541          | 0.963    |
-| No log        | 3.0   | 114  | 0.1495          | 0.9645   |
 ### 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:
+- Loss: 0.1695
+- Accuracy: 0.9445
 ## Model description
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
+| 0.0243        | 1.0   | 625   | 0.2303          | 0.9205   |
+| 0.0341       | 2.0   | 1250   | 0.1541          | 0.933    |
+| 0.0244        | 3.0   | 1875  | 0.1495          | 0.9445   |
 ### Framework versions