emkessle commited on
Commit
4e82999
·
verified ·
1 Parent(s): 89ff80f

End of training

Browse files
Files changed (1) hide show
  1. README.md +9 -9
README.md CHANGED
@@ -21,11 +21,11 @@ should probably proofread and complete it, then remove this comment. -->
21
 
22
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the None dataset.
23
  It achieves the following results on the evaluation set:
24
- - Loss: 0.0002
25
- - Accuracy: 1.0
26
- - F1: 1.0
27
- - Precision: 1.0
28
- - Recall: 1.0
29
 
30
  ## Model description
31
 
@@ -44,7 +44,7 @@ More information needed
44
  ### Training hyperparameters
45
 
46
  The following hyperparameters were used during training:
47
- - learning_rate: 0.0003
48
  - train_batch_size: 8
49
  - eval_batch_size: 8
50
  - seed: 42
@@ -56,9 +56,9 @@ The following hyperparameters were used during training:
56
 
57
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Precision | Recall |
58
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
59
- | 0.0804 | 1.0 | 120 | 0.0662 | 0.9875 | 0.9875 | 0.9878 | 0.9875 |
60
- | 0.0003 | 2.0 | 240 | 0.0002 | 1.0 | 1.0 | 1.0 | 1.0 |
61
- | 0.0002 | 3.0 | 360 | 0.0001 | 1.0 | 1.0 | 1.0 | 1.0 |
62
 
63
 
64
  ### Framework versions
 
21
 
22
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the None dataset.
23
  It achieves the following results on the evaluation set:
24
+ - Loss: 0.1321
25
+ - Accuracy: 0.97
26
+ - F1: 0.9700
27
+ - Precision: 0.9717
28
+ - Recall: 0.97
29
 
30
  ## Model description
31
 
 
44
  ### Training hyperparameters
45
 
46
  The following hyperparameters were used during training:
47
+ - learning_rate: 0.0004
48
  - train_batch_size: 8
49
  - eval_batch_size: 8
50
  - seed: 42
 
56
 
57
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Precision | Recall |
58
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
59
+ | 0.1276 | 1.0 | 120 | 0.2365 | 0.9458 | 0.9457 | 0.9511 | 0.9458 |
60
+ | 0.4081 | 2.0 | 240 | 0.2115 | 0.9583 | 0.9583 | 0.9615 | 0.9583 |
61
+ | 0.1085 | 3.0 | 360 | 0.1289 | 0.9708 | 0.9708 | 0.9724 | 0.9708 |
62
 
63
 
64
  ### Framework versions