ShreyaR commited on
Commit
e1eda83
1 Parent(s): fc11b2e

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +9 -12
README.md CHANGED
@@ -1,6 +1,4 @@
1
  ---
2
- widget:
3
- - text: "I feel lonely. I'm not sure if I will be fine. But I'm trying."
4
  license: mit
5
  tags:
6
  - generated_from_trainer
@@ -18,13 +16,12 @@ should probably proofread and complete it, then remove this comment. -->
18
 
19
  This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
- - Loss: 0.1580
22
- - Accuracy: 0.969
23
 
24
  ## Model description
25
 
26
- - Label 0: Non-Depressive
27
- - Label 1: Depressive
28
 
29
  ## Intended uses & limitations
30
 
@@ -51,14 +48,14 @@ The following hyperparameters were used during training:
51
 
52
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
53
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
54
- | No log | 1.0 | 188 | 0.1907 | 0.965 |
55
- | No log | 2.0 | 376 | 0.1912 | 0.966 |
56
- | 0.1711 | 3.0 | 564 | 0.1580 | 0.969 |
57
 
58
 
59
  ### Framework versions
60
 
61
- - Transformers 4.16.2
62
  - Pytorch 1.10.0+cu111
63
- - Datasets 1.18.3
64
- - Tokenizers 0.11.0
 
1
  ---
 
 
2
  license: mit
3
  tags:
4
  - generated_from_trainer
 
16
 
17
  This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on an unknown dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 0.1385
20
+ - Accuracy: 0.9745
21
 
22
  ## Model description
23
 
24
+ More information needed
 
25
 
26
  ## Intended uses & limitations
27
 
 
48
 
49
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
50
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
51
+ | 0.0238 | 1.0 | 625 | 0.1385 | 0.9745 |
52
+ | 0.0333 | 2.0 | 1250 | 0.1385 | 0.9745 |
53
+ | 0.0263 | 3.0 | 1875 | 0.1385 | 0.9745 |
54
 
55
 
56
  ### Framework versions
57
 
58
+ - Transformers 4.17.0
59
  - Pytorch 1.10.0+cu111
60
+ - Datasets 2.0.0
61
+ - Tokenizers 0.11.6