Hugging Face's logo

Sharpaxis
/
News_classification_distilbert

Text Classification
Transformers
TensorBoard
Safetensors
distilbert
Generated from Trainer
Model card Files
xet
Metrics Community
Sharpaxis commited on
Commit
710f97f
·
verified ·
1 Parent(s): b10fe39

DONE

Browse files
Files changed (4) hide show
  1. README.md +7 -7
  2. model.safetensors +1 -1
  3. runs/Dec06_18-24-17_b8e3dcb04ed1/events.out.tfevents.1733509458.b8e3dcb04ed1.23.1 +3 -0
  4. training_args.bin +1 -1
README.md CHANGED
@@ -18,9 +18,9 @@ should probably proofread and complete it, then remove this comment. -->
18
 
19
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
- - Loss: 0.1243
22
- - F1: 0.9757
23
- - Acc: 0.9765
24
 
25
  ## Model description
26
 
@@ -39,7 +39,7 @@ More information needed
39
  ### Training hyperparameters
40
 
41
  The following hyperparameters were used during training:
42
- - learning_rate: 2e-05
43
  - train_batch_size: 16
44
  - eval_batch_size: 16
45
  - seed: 42
@@ -52,9 +52,9 @@ The following hyperparameters were used during training:
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | F1 | Acc |
54
  |:-------------:|:-----:|:-----:|:---------------:|:------:|:------:|
55
- | 0.1211 | 1.0 | 3868 | 0.0839 | 0.9736 | 0.9746 |
56
- | 0.0506 | 2.0 | 7736 | 0.0941 | 0.9746 | 0.9756 |
57
- | 0.0227 | 3.0 | 11604 | 0.1243 | 0.9757 | 0.9765 |
58
 
59
 
60
  ### Framework versions
 
18
 
19
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 0.1001
22
+ - F1: 0.9786
23
+ - Acc: 0.9795
24
 
25
  ## Model description
26
 
 
39
  ### Training hyperparameters
40
 
41
  The following hyperparameters were used during training:
42
+ - learning_rate: 1e-05
43
  - train_batch_size: 16
44
  - eval_batch_size: 16
45
  - seed: 42
 
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | F1 | Acc |
54
  |:-------------:|:-----:|:-----:|:---------------:|:------:|:------:|
55
+ | 0.1312 | 1.0 | 3868 | 0.0919 | 0.9760 | 0.9771 |
56
+ | 0.063 | 2.0 | 7736 | 0.0884 | 0.9785 | 0.9794 |
57
+ | 0.0396 | 3.0 | 11604 | 0.1001 | 0.9786 | 0.9795 |
58
 
59
 
60
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5e2c77955968595ea78a35e22e59611174278c0148e54e43e0b53844d1ec2bf8
3
  size 267832560
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:833087bbdda05da41cb99ecfb1c55805cb8a7d784cd1cf58fe0e6b5e3c384ae2
3
  size 267832560
runs/Dec06_18-24-17_b8e3dcb04ed1/events.out.tfevents.1733509458.b8e3dcb04ed1.23.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:07cef5753f495919bc9d1a08c1c5deac11f7ff503c9b9d8647f5c02c4e53ae78
3
+ size 7059
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:796cd83fa20e93e2e70a5c9357581b609923e767ef301656e6557837e61de4ea
3
  size 5304
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8f8edef5bd5699559ba0b8e3b8d1cd9a64284e457a6c3f9f0a593b866b8ec727
3
  size 5304