DTempo commited on
Commit
b2261e5
·
1 Parent(s): 6b35a16

End of training

Browse files
Files changed (2) hide show
  1. README.md +80 -0
  2. adapter_model.safetensors +1 -1
README.md ADDED
@@ -0,0 +1,80 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: bert-base-uncased
4
+ tags:
5
+ - generated_from_trainer
6
+ metrics:
7
+ - accuracy
8
+ - precision
9
+ - recall
10
+ - f1
11
+ model-index:
12
+ - name: whatsapp-group-classifierv2
13
+ results: []
14
+ ---
15
+
16
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
17
+ should probably proofread and complete it, then remove this comment. -->
18
+
19
+ # whatsapp-group-classifierv2
20
+
21
+ This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on the None dataset.
22
+ It achieves the following results on the evaluation set:
23
+ - Loss: 0.4619
24
+ - Accuracy: 0.8283
25
+ - Precision: 0.8483
26
+ - Recall: 0.8347
27
+ - F1: 0.8407
28
+
29
+ ## Model description
30
+
31
+ More information needed
32
+
33
+ ## Intended uses & limitations
34
+
35
+ More information needed
36
+
37
+ ## Training and evaluation data
38
+
39
+ More information needed
40
+
41
+ ## Training procedure
42
+
43
+ ### Training hyperparameters
44
+
45
+ The following hyperparameters were used during training:
46
+ - learning_rate: 4e-05
47
+ - train_batch_size: 16
48
+ - eval_batch_size: 16
49
+ - seed: 42
50
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
51
+ - lr_scheduler_type: linear
52
+ - num_epochs: 15
53
+
54
+ ### Training results
55
+
56
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision | Recall | F1 |
57
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|:---------:|:------:|:------:|
58
+ | 0.9676 | 1.0 | 513 | 0.7151 | 0.7322 | 0.7315 | 0.7148 | 0.7182 |
59
+ | 0.6846 | 2.0 | 1026 | 0.5747 | 0.7741 | 0.7918 | 0.7716 | 0.7732 |
60
+ | 0.5657 | 3.0 | 1539 | 0.5265 | 0.7946 | 0.8224 | 0.7882 | 0.7983 |
61
+ | 0.5294 | 4.0 | 2052 | 0.4870 | 0.8059 | 0.8260 | 0.8084 | 0.8163 |
62
+ | 0.4941 | 5.0 | 2565 | 0.4787 | 0.8102 | 0.8326 | 0.8122 | 0.8210 |
63
+ | 0.4956 | 6.0 | 3078 | 0.4796 | 0.8156 | 0.8363 | 0.8183 | 0.8252 |
64
+ | 0.4701 | 7.0 | 3591 | 0.4812 | 0.8146 | 0.8370 | 0.8165 | 0.8230 |
65
+ | 0.4521 | 8.0 | 4104 | 0.4738 | 0.8220 | 0.8398 | 0.8287 | 0.8336 |
66
+ | 0.444 | 9.0 | 4617 | 0.4664 | 0.8195 | 0.8412 | 0.8240 | 0.8313 |
67
+ | 0.4336 | 10.0 | 5130 | 0.4682 | 0.8244 | 0.8445 | 0.8311 | 0.8370 |
68
+ | 0.4345 | 11.0 | 5643 | 0.4653 | 0.8229 | 0.8446 | 0.8285 | 0.8351 |
69
+ | 0.4275 | 12.0 | 6156 | 0.4619 | 0.8254 | 0.8463 | 0.8322 | 0.8382 |
70
+ | 0.4155 | 13.0 | 6669 | 0.4620 | 0.8288 | 0.8487 | 0.8356 | 0.8414 |
71
+ | 0.4179 | 14.0 | 7182 | 0.4624 | 0.8283 | 0.8485 | 0.8354 | 0.8412 |
72
+ | 0.4078 | 15.0 | 7695 | 0.4619 | 0.8283 | 0.8483 | 0.8347 | 0.8407 |
73
+
74
+
75
+ ### Framework versions
76
+
77
+ - Transformers 4.35.2
78
+ - Pytorch 2.1.0+cu118
79
+ - Datasets 2.15.0
80
+ - Tokenizers 0.15.0
adapter_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e9793884f3ffd6531e2c37fb48a3d9170989752d60ee5295ab45016861022ede
3
  size 2378568
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a2956964bc70c0b0c908000377c9938f95090047c1d8ebfec806c10194234db4
3
  size 2378568