Hugging Face's logo

Thebisso09
/
final_model

Text Classification
Transformers
TensorBoard
Safetensors
distilbert
Generated from Trainer
text-embeddings-inference
Model card Files
xet
 Metrics Community
Thebisso09 commited on
Commit
2ca60a6
·
verified ·
1 Parent(s): e54db74

YourModelName

Browse files
Files changed (2) hide show
  1. README.md +12 -10
  2. model.safetensors +1 -1
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-uncased](https://huggingface.co/distilbert-base-uncased) on the None dataset.
22
  It achieves the following results on the evaluation set:
23
- - Loss: 0.6844
24
- - Accuracy: 0.8312
25
- - Precision: 0.8805
26
- - Recall: 0.9256
27
- - F1: 0.9025
28
 
29
  ## Model description
30
 
@@ -48,17 +48,19 @@ The following hyperparameters were used during training:
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: 10
53
 
54
  ### Training results
55
 
56
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision | Recall | F1 |
57
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:---------:|:------:|:------:|
58
- | No log | 1.0 | 347 | 0.4960 | 0.8463 | 0.8730 | 0.9573 | 0.9132 |
59
- | 0.268 | 2.0 | 694 | 0.5739 | 0.8413 | 0.8794 | 0.9410 | 0.9092 |
60
- | 0.2435 | 3.0 | 1041 | 0.6603 | 0.8333 | 0.8747 | 0.9368 | 0.9047 |
61
- | 0.2435 | 4.0 | 1388 | 0.6844 | 0.8312 | 0.8805 | 0.9256 | 0.9025 |
 
 
62
 
63
 
64
  ### Framework versions
 
20
 
21
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the None dataset.
22
  It achieves the following results on the evaluation set:
23
+ - Loss: 0.7222
24
+ - Accuracy: 0.8009
25
+ - Precision: 0.8760
26
+ - Recall: 0.8851
27
+ - F1: 0.8805
28
 
29
  ## Model description
30
 
 
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: reduce_lr_on_plateau
52
  - num_epochs: 10
53
 
54
  ### Training results
55
 
56
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision | Recall | F1 |
57
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:---------:|:------:|:------:|
58
+ | No log | 1.0 | 347 | 0.4389 | 0.8333 | 0.8425 | 0.9826 | 0.9072 |
59
+ | 0.3625 | 2.0 | 694 | 0.4550 | 0.8398 | 0.8503 | 0.9791 | 0.9102 |
60
+ | 0.3124 | 3.0 | 1041 | 0.4808 | 0.8175 | 0.8596 | 0.9321 | 0.8944 |
61
+ | 0.3124 | 4.0 | 1388 | 0.5418 | 0.8268 | 0.8554 | 0.9521 | 0.9012 |
62
+ | 0.2597 | 5.0 | 1735 | 0.5752 | 0.8261 | 0.8525 | 0.9556 | 0.9011 |
63
+ | 0.2365 | 6.0 | 2082 | 0.7222 | 0.8009 | 0.8760 | 0.8851 | 0.8805 |
64
 
65
 
66
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:dc5088db5d1a7e3cead6523ddf6a7937d4a611e26f38e9e53ba051c558591cbf
3
  size 267832560
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7e495ca39979280e396e338c62e3d61026b1c904b10035d927341fa93bed3c61
3
  size 267832560