jpbywater commited on
Commit
d2cb8b6
·
verified ·
1 Parent(s): ef2658d

End of training

Browse files
Files changed (2) hide show
  1. README.md +10 -8
  2. model.safetensors +1 -1
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/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on an unknown dataset.
23
  It achieves the following results on the evaluation set:
24
- - Loss: 0.4238
25
- - Accuracy: 0.8454
26
- - F1: 0.8386
27
- - Precision: 0.8510
28
- - Recall: 0.8454
29
 
30
  ## Model description
31
 
@@ -50,14 +50,16 @@ The following hyperparameters were used during training:
50
  - seed: 42
51
  - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
52
  - lr_scheduler_type: linear
53
- - num_epochs: 2
54
 
55
  ### Training results
56
 
57
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Precision | Recall |
58
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
59
- | No log | 1.0 | 164 | 0.4494 | 0.8316 | 0.8219 | 0.8365 | 0.8316 |
60
- | No log | 2.0 | 328 | 0.4238 | 0.8454 | 0.8386 | 0.8510 | 0.8454 |
 
 
61
 
62
 
63
  ### Framework versions
 
21
 
22
  This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on an unknown dataset.
23
  It achieves the following results on the evaluation set:
24
+ - Loss: 0.4615
25
+ - Accuracy: 0.8522
26
+ - F1: 0.8506
27
+ - Precision: 0.8536
28
+ - Recall: 0.8522
29
 
30
  ## Model description
31
 
 
50
  - seed: 42
51
  - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
52
  - lr_scheduler_type: linear
53
+ - num_epochs: 4
54
 
55
  ### Training results
56
 
57
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Precision | Recall |
58
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
59
+ | No log | 1.0 | 164 | 0.4292 | 0.8247 | 0.8266 | 0.8292 | 0.8247 |
60
+ | No log | 2.0 | 328 | 0.4365 | 0.8351 | 0.8314 | 0.8334 | 0.8351 |
61
+ | No log | 3.0 | 492 | 0.4568 | 0.8385 | 0.8395 | 0.8416 | 0.8385 |
62
+ | 0.2652 | 4.0 | 656 | 0.4615 | 0.8522 | 0.8506 | 0.8536 | 0.8522 |
63
 
64
 
65
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:27a9fc1f4040794264d6f7c1f6a21a30bbb985bd8113a0247e204d7e73525b10
3
  size 267838720
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b9eb5e7464b407f913b195204443d4d7bd00811087403428ba7304ed5ffd7f6d
3
  size 267838720