Hugging Face's logo

SofieElving
/
classifier-chapter4

Text Classification
Transformers
TensorBoard
Safetensors
distilbert
Generated from Trainer
text-embeddings-inference
Model card Files
xet
Metrics Community
SofieElving commited on
Commit
bf1e4fe
·
verified ·
1 Parent(s): 7f1ffa8

End of training

Browse files
Files changed (7) hide show
  1. README.md +6 -6
  2. config.json +8 -8
  3. model.safetensors +1 -1
  4. runs/Oct24_12-19-17_dd37e4074e0b/events.out.tfevents.1761308363.dd37e4074e0b.41432.0 +3 -0
  5. runs/Oct24_13-57-05_dd37e4074e0b/events.out.tfevents.1761314232.dd37e4074e0b.50070.6 +3 -0
  6. runs/Oct24_13-57-05_dd37e4074e0b/events.out.tfevents.1761314629.dd37e4074e0b.50070.7 +3 -0
  7. training_args.bin +1 -1
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.2540
23
- - Accuracy: 0.9228
24
- - F1: 0.9227
25
 
26
  ## Model description
27
 
@@ -52,9 +52,9 @@ The following hyperparameters were used during training:
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
54
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
55
- | No log | 1.0 | 313 | 0.2672 | 0.9133 | 0.9133 |
56
- | 0.3134 | 2.0 | 626 | 0.2472 | 0.9205 | 0.9204 |
57
- | 0.3134 | 3.0 | 939 | 0.2540 | 0.9228 | 0.9227 |
58
 
59
 
60
  ### Framework versions
 
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.2537
23
+ - Accuracy: 0.9242
24
+ - F1: 0.9243
25
 
26
  ## Model description
27
 
 
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 |
54
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|
55
+ | No log | 1.0 | 313 | 0.2693 | 0.9083 | 0.9082 |
56
+ | 0.3113 | 2.0 | 626 | 0.2394 | 0.9238 | 0.9237 |
57
+ | 0.3113 | 3.0 | 939 | 0.2537 | 0.9242 | 0.9243 |
58
 
59
 
60
  ### Framework versions
config.json CHANGED
@@ -9,17 +9,17 @@
9
  "dtype": "float32",
10
  "hidden_dim": 3072,
11
  "id2label": {
12
- "0": "World",
13
- "1": "Sports",
14
- "2": "Business",
15
- "3": "Sci/Tech"
16
  },
17
  "initializer_range": 0.02,
18
  "label2id": {
19
- "World": 0,
20
- "Sports": 1,
21
- "Business": 2,
22
- "Sci/Tech": 3
23
  },
24
  "max_position_embeddings": 512,
25
  "model_type": "distilbert",
 
9
  "dtype": "float32",
10
  "hidden_dim": 3072,
11
  "id2label": {
12
+ "0": "LABEL_0",
13
+ "1": "LABEL_1",
14
+ "2": "LABEL_2",
15
+ "3": "LABEL_3"
16
  },
17
  "initializer_range": 0.02,
18
  "label2id": {
19
+ "LABEL_0": 0,
20
+ "LABEL_1": 1,
21
+ "LABEL_2": 2,
22
+ "LABEL_3": 3
23
  },
24
  "max_position_embeddings": 512,
25
  "model_type": "distilbert",
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:029b8e40267b436593a6466dc120a1c645671d326b39e416066c84b0304e20ce
3
  size 267838720
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9a4e405df62e32090f907225ae68955615f3ecb4ea2b3e6aa547fecac63ccef6
3
  size 267838720
runs/Oct24_12-19-17_dd37e4074e0b/events.out.tfevents.1761308363.dd37e4074e0b.41432.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bcdaae2af573ebc690cf5cfdaefb7b9430e98f78d51d5816c56d3b81074f45eb
3
+ size 6771
runs/Oct24_13-57-05_dd37e4074e0b/events.out.tfevents.1761314232.dd37e4074e0b.50070.6 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5344b5c44b1dd473ab04b56384ebead38cadec84caff17cd2a736ddbf61fa7d0
3
+ size 6771
runs/Oct24_13-57-05_dd37e4074e0b/events.out.tfevents.1761314629.dd37e4074e0b.50070.7 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:535dbec9ab27817a3caf0526aaf0e8de158311c9203c5b70b8ce393bae7d5e1b
3
+ size 457
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8e3725f44a2bd2e8d2f1b87be9ac13c99c0009ea1575aabe623128b03f1413fb
3
  size 5841
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0903b7ff323b05a8c19e96da9eb20cfb8579434c99818efe29cfb25b5c899eba
3
  size 5841