brucedkyle commited on
Commit
113efbf
·
verified ·
1 Parent(s): 7ee1a43

Training completed!

Browse files
Files changed (1) hide show
  1. README.md +6 -6
README.md CHANGED
@@ -19,9 +19,9 @@ should probably proofread and complete it, then remove this comment. -->
19
 
20
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
21
  It achieves the following results on the evaluation set:
22
- - Loss: 0.2105
23
- - Accuracy: 0.9255
24
- - F1: 0.9256
25
 
26
  ## Model description
27
 
@@ -52,13 +52,13 @@ The following hyperparameters were used during training:
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
54
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
55
- | 0.799 | 1.0 | 250 | 0.2985 | 0.91 | 0.9095 |
56
- | 0.2437 | 2.0 | 500 | 0.2105 | 0.9255 | 0.9256 |
57
 
58
 
59
  ### Framework versions
60
 
61
  - Transformers 4.44.2
62
- - Pytorch 2.4.0.post301
63
  - Datasets 2.19.1
64
  - Tokenizers 0.19.1
 
19
 
20
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
21
  It achieves the following results on the evaluation set:
22
+ - Loss: 0.2124
23
+ - Accuracy: 0.926
24
+ - F1: 0.9259
25
 
26
  ## Model description
27
 
 
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
54
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
55
+ | 0.8302 | 1.0 | 250 | 0.3161 | 0.9085 | 0.9083 |
56
+ | 0.2489 | 2.0 | 500 | 0.2124 | 0.926 | 0.9259 |
57
 
58
 
59
  ### Framework versions
60
 
61
  - Transformers 4.44.2
62
+ - Pytorch 2.4.0
63
  - Datasets 2.19.1
64
  - Tokenizers 0.19.1