Hugging Face's logo

ziq
/
depression_suggestion

Text Generation
Transformers
PyTorch
TensorBoard
gpt2
Generated from Trainer
text-generation-inference
Inference Endpoints
Model card Files Metrics Community
1
ziq commited on
Commit
aa16c86
·
1 Parent(s): 3187b8e

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +126 -0
README.md ADDED
@@ -0,0 +1,126 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - generated_from_trainer
5
+ model-index:
6
+ - name: depression_suggestion
7
+ results: []
8
+ ---
9
+
10
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
11
+ should probably proofread and complete it, then remove this comment. -->
12
+
13
+ # depression_suggestion
14
+
15
+ This model is a fine-tuned version of [distilgpt2](https://huggingface.co/distilgpt2) on an unknown dataset.
16
+ It achieves the following results on the evaluation set:
17
+ - Loss: 3.3740
18
+
19
+ ## Model description
20
+
21
+ More information needed
22
+
23
+ ## Intended uses & limitations
24
+
25
+ More information needed
26
+
27
+ ## Training and evaluation data
28
+
29
+ More information needed
30
+
31
+ ## Training procedure
32
+
33
+ ### Training hyperparameters
34
+
35
+ The following hyperparameters were used during training:
36
+ - learning_rate: 5e-05
37
+ - train_batch_size: 32
38
+ - eval_batch_size: 64
39
+ - seed: 42
40
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
41
+ - lr_scheduler_type: linear
42
+ - lr_scheduler_warmup_steps: 500
43
+ - num_epochs: 70
44
+
45
+ ### Training results
46
+
47
+ | Training Loss | Epoch | Step | Validation Loss |
48
+ |:-------------:|:-----:|:----:|:---------------:|
49
+ | No log | 1.0 | 3 | 60.7965 |
50
+ | No log | 2.0 | 6 | 60.5778 |
51
+ | No log | 3.0 | 9 | 60.1954 |
52
+ | No log | 4.0 | 12 | 59.6487 |
53
+ | No log | 5.0 | 15 | 58.9372 |
54
+ | No log | 6.0 | 18 | 58.0582 |
55
+ | No log | 7.0 | 21 | 57.0106 |
56
+ | No log | 8.0 | 24 | 55.7910 |
57
+ | No log | 9.0 | 27 | 54.3934 |
58
+ | No log | 10.0 | 30 | 52.8099 |
59
+ | No log | 11.0 | 33 | 51.0219 |
60
+ | No log | 12.0 | 36 | 49.0127 |
61
+ | No log | 13.0 | 39 | 46.7522 |
62
+ | No log | 14.0 | 42 | 44.2033 |
63
+ | No log | 15.0 | 45 | 41.3146 |
64
+ | No log | 16.0 | 48 | 37.9982 |
65
+ | No log | 17.0 | 51 | 34.2236 |
66
+ | No log | 18.0 | 54 | 29.8068 |
67
+ | No log | 19.0 | 57 | 24.9750 |
68
+ | No log | 20.0 | 60 | 20.0707 |
69
+ | No log | 21.0 | 63 | 15.5166 |
70
+ | No log | 22.0 | 66 | 12.0328 |
71
+ | No log | 23.0 | 69 | 9.1012 |
72
+ | No log | 24.0 | 72 | 7.2116 |
73
+ | No log | 25.0 | 75 | 6.3149 |
74
+ | No log | 26.0 | 78 | 5.8127 |
75
+ | No log | 27.0 | 81 | 5.4548 |
76
+ | No log | 28.0 | 84 | 5.1684 |
77
+ | No log | 29.0 | 87 | 4.8927 |
78
+ | No log | 30.0 | 90 | 4.6128 |
79
+ | No log | 31.0 | 93 | 4.3782 |
80
+ | No log | 32.0 | 96 | 4.1996 |
81
+ | No log | 33.0 | 99 | 4.0981 |
82
+ | No log | 34.0 | 102 | 4.0022 |
83
+ | No log | 35.0 | 105 | 3.9224 |
84
+ | No log | 36.0 | 108 | 3.8381 |
85
+ | No log | 37.0 | 111 | 3.7660 |
86
+ | No log | 38.0 | 114 | 3.6887 |
87
+ | No log | 39.0 | 117 | 3.6483 |
88
+ | No log | 40.0 | 120 | 3.6020 |
89
+ | No log | 41.0 | 123 | 3.5590 |
90
+ | No log | 42.0 | 126 | 3.5199 |
91
+ | No log | 43.0 | 129 | 3.4646 |
92
+ | No log | 44.0 | 132 | 3.4098 |
93
+ | No log | 45.0 | 135 | 3.3684 |
94
+ | No log | 46.0 | 138 | 3.3290 |
95
+ | No log | 47.0 | 141 | 3.3113 |
96
+ | No log | 48.0 | 144 | 3.3033 |
97
+ | No log | 49.0 | 147 | 3.2928 |
98
+ | No log | 50.0 | 150 | 3.2776 |
99
+ | No log | 51.0 | 153 | 3.2587 |
100
+ | No log | 52.0 | 156 | 3.2487 |
101
+ | No log | 53.0 | 159 | 3.2390 |
102
+ | No log | 54.0 | 162 | 3.2318 |
103
+ | No log | 55.0 | 165 | 3.2311 |
104
+ | No log | 56.0 | 168 | 3.2377 |
105
+ | No log | 57.0 | 171 | 3.2554 |
106
+ | No log | 58.0 | 174 | 3.2720 |
107
+ | No log | 59.0 | 177 | 3.2781 |
108
+ | No log | 60.0 | 180 | 3.2882 |
109
+ | No log | 61.0 | 183 | 3.3089 |
110
+ | No log | 62.0 | 186 | 3.3352 |
111
+ | No log | 63.0 | 189 | 3.3519 |
112
+ | No log | 64.0 | 192 | 3.3233 |
113
+ | No log | 65.0 | 195 | 3.3028 |
114
+ | No log | 66.0 | 198 | 3.3153 |
115
+ | No log | 67.0 | 201 | 3.3422 |
116
+ | No log | 68.0 | 204 | 3.3753 |
117
+ | No log | 69.0 | 207 | 3.4003 |
118
+ | No log | 70.0 | 210 | 3.3740 |
119
+
120
+
121
+ ### Framework versions
122
+
123
+ - Transformers 4.19.2
124
+ - Pytorch 1.11.0+cu113
125
+ - Datasets 2.2.2
126
+ - Tokenizers 0.12.1