Trong-Nghia commited on
Commit
1f4878b
1 Parent(s): 4ff56f8

End of training

Browse files
README.md ADDED
@@ -0,0 +1,63 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ base_model: roberta-large
4
+ tags:
5
+ - generated_from_trainer
6
+ metrics:
7
+ - accuracy
8
+ model-index:
9
+ - name: roberta-large-depression-classification-v2
10
+ results: []
11
+ ---
12
+
13
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
14
+ should probably proofread and complete it, then remove this comment. -->
15
+
16
+ # roberta-large-depression-classification-v2
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.2328
21
+ - Accuracy: 0.5435
22
+ - F1 Score: 0.5316
23
+
24
+ ## Model description
25
+
26
+ More information needed
27
+
28
+ ## Intended uses & limitations
29
+
30
+ More information needed
31
+
32
+ ## Training and evaluation data
33
+
34
+ More information needed
35
+
36
+ ## Training procedure
37
+
38
+ ### Training hyperparameters
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: 3
48
+
49
+ ### Training results
50
+
51
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 Score |
52
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|:--------:|
53
+ | 0.9778 | 1.0 | 677 | 1.2323 | 0.5380 | 0.5275 |
54
+ | 0.6377 | 2.0 | 1354 | 2.0223 | 0.5315 | 0.5125 |
55
+ | 0.5285 | 3.0 | 2031 | 2.2328 | 0.5435 | 0.5316 |
56
+
57
+
58
+ ### Framework versions
59
+
60
+ - Transformers 4.31.0
61
+ - Pytorch 2.0.1+cu118
62
+ - Datasets 2.14.4
63
+ - Tokenizers 0.13.3
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:183d4ef6ca76bebcf0419414ef22a8996553e05d23318ab9d21cac2422bd9bbe
3
  size 1421586865
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:be4044a99c67bebb51f19447fd2feaffef3efe119803d586e85df8757d945ab5
3
  size 1421586865
runs/Aug21_08-21-21_95eb6f6307b9/events.out.tfevents.1692606129.95eb6f6307b9.457.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e15a51a0e5e627f55f68ce92d60a58d7893de18a345cdd1fb075762d9383d76b
3
- size 5742
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:82e2c58306b2ed30a173079b76dbdc72a6575da33b0c510b0b18b132d10a5159
3
+ size 6471