Hugging Face's logo

Trong-Nghia
/
roberta-large-depression-classification

Text Classification
Transformers
PyTorch
TensorBoard
roberta
Generated from Trainer
text-embeddings-inference
Model card Files
xet
 Metrics Community
1
Trong-Nghia commited on
Commit
6afc515
·
1 Parent(s): fc26d59

End of training

Browse files
Files changed (3) hide show
  1. README.md +8 -7
  2. pytorch_model.bin +1 -1
  3. runs/Aug21_05-20-53_1b11b05bca4c/events.out.tfevents.1692595634.1b11b05bca4c.430.0 +2 -2
README.md CHANGED
@@ -17,9 +17,9 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  This model is a fine-tuned version of [roberta-large](https://huggingface.co/roberta-large) on the None dataset.
19
  It achieves the following results on the evaluation set:
20
- - Loss: 1.1247
21
- - Accuracy: 0.5841
22
- - F1 Score: 0.5479
23
 
24
  ## Model description
25
 
@@ -39,18 +39,19 @@ More information needed
39
 
40
  The following hyperparameters were used during training:
41
  - learning_rate: 5e-06
42
- - train_batch_size: 8
43
- - eval_batch_size: 8
44
  - seed: 42
45
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
46
  - lr_scheduler_type: linear
47
- - num_epochs: 1
48
 
49
  ### Training results
50
 
51
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 Score |
52
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:--------:|
53
- | 0.5166 | 1.0 | 1112 | 1.1247 | 0.5841 | 0.5479 |
 
54
 
55
 
56
  ### Framework versions
 
17
 
18
  This model is a fine-tuned version of [roberta-large](https://huggingface.co/roberta-large) on the None dataset.
19
  It achieves the following results on the evaluation set:
20
+ - Loss: 2.4876
21
+ - Accuracy: 0.5558
22
+ - F1 Score: 0.5450
23
 
24
  ## Model description
25
 
 
39
 
40
  The following hyperparameters were used during training:
41
  - learning_rate: 5e-06
42
+ - train_batch_size: 4
43
+ - eval_batch_size: 4
44
  - seed: 42
45
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
46
  - lr_scheduler_type: linear
47
+ - num_epochs: 2
48
 
49
  ### Training results
50
 
51
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 Score |
52
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:--------:|
53
+ | 0.6381 | 1.0 | 2223 | 1.9208 | 0.5423 | 0.5413 |
54
+ | 0.4746 | 2.0 | 4446 | 2.4876 | 0.5558 | 0.5450 |
55
 
56
 
57
  ### Framework versions
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6af859d97295867c30cb7069171da36a3126c1c40f4e0122ceeec91499e7cec5
3
  size 1421586865
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:930a1f00a15ca1f95c07dece1e92cd6ade049f857ab416b676047289091e0c46
3
  size 1421586865
runs/Aug21_05-20-53_1b11b05bca4c/events.out.tfevents.1692595634.1b11b05bca4c.430.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:643b39a008675882e3aea8b6b290cff715c71f060a32c0eb35e48b5bc9f3a82f
3
- size 5983
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:38726034515747a8633e800b0b1b001e07c7b6e966148b6d514021d9beea9600
3
+ size 6712