Raj-Sharma commited on
Commit
943ecf8
·
verified ·
1 Parent(s): 52d302c

Training complete

Browse files
Files changed (1) hide show
  1. README.md +16 -12
README.md CHANGED
@@ -20,11 +20,11 @@ should probably proofread and complete it, then remove this comment. -->
20
 
21
  This model is a fine-tuned version of [distilbert-base-cased](https://huggingface.co/distilbert-base-cased) on the None dataset.
22
  It achieves the following results on the evaluation set:
23
- - Loss: 0.0330
24
- - Precision: 0.9467
25
- - Recall: 0.9645
26
- - F1: 0.9555
27
- - Accuracy: 0.9903
28
 
29
  ## Model description
30
 
@@ -43,21 +43,25 @@ More information needed
43
  ### Training hyperparameters
44
 
45
  The following hyperparameters were used during training:
46
- - learning_rate: 5e-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: 3
53
 
54
  ### Training results
55
 
56
- | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
57
- |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
58
- | 0.0464 | 1.0 | 2180 | 0.0397 | 0.9267 | 0.9565 | 0.9413 | 0.9875 |
59
- | 0.0263 | 2.0 | 4360 | 0.0340 | 0.9400 | 0.9608 | 0.9503 | 0.9894 |
60
- | 0.0156 | 3.0 | 6540 | 0.0330 | 0.9467 | 0.9645 | 0.9555 | 0.9903 |
 
 
 
 
61
 
62
 
63
  ### Framework versions
 
20
 
21
  This model is a fine-tuned version of [distilbert-base-cased](https://huggingface.co/distilbert-base-cased) on the None dataset.
22
  It achieves the following results on the evaluation set:
23
+ - Loss: 0.0517
24
+ - Precision: 0.9468
25
+ - Recall: 0.9623
26
+ - F1: 0.9545
27
+ - Accuracy: 0.9908
28
 
29
  ## Model description
30
 
 
43
  ### Training hyperparameters
44
 
45
  The following hyperparameters were used during training:
46
+ - learning_rate: 7e-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: 7
53
 
54
  ### Training results
55
 
56
+ | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
57
+ |:-------------:|:-----:|:-----:|:---------------:|:---------:|:------:|:------:|:--------:|
58
+ | 0.047 | 1.0 | 2180 | 0.0385 | 0.9149 | 0.9466 | 0.9305 | 0.9879 |
59
+ | 0.0308 | 2.0 | 4360 | 0.0350 | 0.9181 | 0.9521 | 0.9348 | 0.9887 |
60
+ | 0.0199 | 3.0 | 6540 | 0.0334 | 0.9333 | 0.9546 | 0.9438 | 0.9895 |
61
+ | 0.0112 | 4.0 | 8720 | 0.0377 | 0.9412 | 0.9598 | 0.9504 | 0.9901 |
62
+ | 0.0065 | 5.0 | 10900 | 0.0417 | 0.9373 | 0.9619 | 0.9495 | 0.9903 |
63
+ | 0.0033 | 6.0 | 13080 | 0.0445 | 0.9464 | 0.9611 | 0.9537 | 0.9907 |
64
+ | 0.0014 | 7.0 | 15260 | 0.0517 | 0.9468 | 0.9623 | 0.9545 | 0.9908 |
65
 
66
 
67
  ### Framework versions