RavingRabbit commited on
Commit
44765b7
·
1 Parent(s): ada49c6

End of training

Browse files
Files changed (2) hide show
  1. README.md +9 -9
  2. model.safetensors +1 -1
README.md CHANGED
@@ -17,8 +17,8 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
19
  It achieves the following results on the evaluation set:
20
- - Loss: 0.7209
21
- - Matthews Correlation: 0.5461
22
 
23
  ## Model description
24
 
@@ -37,10 +37,10 @@ More information needed
37
  ### Training hyperparameters
38
 
39
  The following hyperparameters were used during training:
40
- - learning_rate: 2e-05
41
  - train_batch_size: 16
42
  - eval_batch_size: 16
43
- - seed: 42
44
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
  - lr_scheduler_type: linear
46
  - num_epochs: 5
@@ -49,11 +49,11 @@ The following hyperparameters were used during training:
49
 
50
  | Training Loss | Epoch | Step | Validation Loss | Matthews Correlation |
51
  |:-------------:|:-----:|:----:|:---------------:|:--------------------:|
52
- | 0.529 | 1.0 | 535 | 0.4688 | 0.4306 |
53
- | 0.36 | 2.0 | 1070 | 0.4656 | 0.5343 |
54
- | 0.2495 | 3.0 | 1605 | 0.6466 | 0.5076 |
55
- | 0.1818 | 4.0 | 2140 | 0.7209 | 0.5461 |
56
- | 0.1304 | 5.0 | 2675 | 0.8472 | 0.5256 |
57
 
58
 
59
  ### Framework versions
 
17
 
18
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
19
  It achieves the following results on the evaluation set:
20
+ - Loss: 0.7844
21
+ - Matthews Correlation: 0.5466
22
 
23
  ## Model description
24
 
 
37
  ### Training hyperparameters
38
 
39
  The following hyperparameters were used during training:
40
+ - learning_rate: 1.6657021403235714e-05
41
  - train_batch_size: 16
42
  - eval_batch_size: 16
43
+ - seed: 10
44
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
  - lr_scheduler_type: linear
46
  - num_epochs: 5
 
49
 
50
  | Training Loss | Epoch | Step | Validation Loss | Matthews Correlation |
51
  |:-------------:|:-----:|:----:|:---------------:|:--------------------:|
52
+ | 0.5213 | 1.0 | 535 | 0.4858 | 0.4326 |
53
+ | 0.3536 | 2.0 | 1070 | 0.4809 | 0.5083 |
54
+ | 0.2508 | 3.0 | 1605 | 0.5551 | 0.5268 |
55
+ | 0.183 | 4.0 | 2140 | 0.7043 | 0.5280 |
56
+ | 0.154 | 5.0 | 2675 | 0.7844 | 0.5466 |
57
 
58
 
59
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:939271f6ad84d6a30c3af3110ecb1a45e8b84279a83c889e1e6f1c15a1782651
3
  size 267832560
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:32e471fba9fc82c67a4b574b100126083bdec7689d71add216b3e83c33f8b78f
3
  size 267832560