mrjunos commited on
Commit
856d3c1
1 Parent(s): 94a0226

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -12
README.md CHANGED
@@ -6,11 +6,6 @@ datasets:
6
  - depression-reddit-cleaned
7
  metrics:
8
  - accuracy
9
- widget:
10
- - text: ["i just found out my boyfriend is depressed i really want to be there for him but i feel like i ve only been saying the wrong thing how can i be there for him help him and see him get better i m worried it will continue to the point it will consume him i can already see his personality changing and i m scared for the future what thing can i say or do to comfort or help"]
11
- example_title: depression
12
- - text: ["i m getting more and more people asking where they can buy the ambients album simple answer is quot not yet quot it ll be on itunes eventually"]
13
- example_title: not_depression
14
  model-index:
15
  - name: depression-reddit-distilroberta-base
16
  results:
@@ -26,7 +21,7 @@ model-index:
26
  metrics:
27
  - name: Accuracy
28
  type: accuracy
29
- value: 0.9831932773109243
30
  ---
31
 
32
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -36,8 +31,8 @@ should probably proofread and complete it, then remove this comment. -->
36
 
37
  This model is a fine-tuned version of [distilroberta-base](https://huggingface.co/distilroberta-base) on the depression-reddit-cleaned dataset.
38
  It achieves the following results on the evaluation set:
39
- - Loss: 0.0948
40
- - Accuracy: 0.9832
41
 
42
  ## Model description
43
 
@@ -68,10 +63,10 @@ The following hyperparameters were used during training:
68
 
69
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
70
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
71
- | 0.1807 | 0.65 | 500 | 0.1087 | 0.9683 |
72
- | 0.1227 | 1.29 | 1000 | 0.1002 | 0.9793 |
73
- | 0.0663 | 1.94 | 1500 | 0.1097 | 0.9806 |
74
- | 0.031 | 2.59 | 2000 | 0.0948 | 0.9832 |
75
 
76
 
77
  ### Framework versions
 
6
  - depression-reddit-cleaned
7
  metrics:
8
  - accuracy
 
 
 
 
 
9
  model-index:
10
  - name: depression-reddit-distilroberta-base
11
  results:
 
21
  metrics:
22
  - name: Accuracy
23
  type: accuracy
24
+ value: 0.9786683904330963
25
  ---
26
 
27
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
31
 
32
  This model is a fine-tuned version of [distilroberta-base](https://huggingface.co/distilroberta-base) on the depression-reddit-cleaned dataset.
33
  It achieves the following results on the evaluation set:
34
+ - Loss: 0.1287
35
+ - Accuracy: 0.9787
36
 
37
  ## Model description
38
 
 
63
 
64
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
65
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
66
+ | 0.1792 | 0.65 | 500 | 0.1206 | 0.9709 |
67
+ | 0.1074 | 1.29 | 1000 | 0.1155 | 0.9716 |
68
+ | 0.0649 | 1.94 | 1500 | 0.0980 | 0.9825 |
69
+ | 0.0243 | 2.59 | 2000 | 0.1287 | 0.9787 |
70
 
71
 
72
  ### Framework versions