Hugging Face's logo

cdc-hf
/
bert-platform-classifier

Text Classification
Transformers
TensorBoard
Safetensors
bert
Generated from Trainer
text-embeddings-inference
Model card Files
xet
 Metrics Community
cdc-hf commited on
Commit
3eb8280
·
verified ·
1 Parent(s): 5222907

End of training

Browse files
Files changed (6) hide show
  1. README.md +12 -12
  2. config.json +4 -2
  3. model.safetensors +2 -2
  4. runs/Oct29_14-25-13_e7ef80ecc11b/events.out.tfevents.1761747921.e7ef80ecc11b.202.11 +3 -0
  5. runs/Oct29_14-30-53_e7ef80ecc11b/events.out.tfevents.1761748257.e7ef80ecc11b.202.12 +3 -0
  6. training_args.bin +1 -1
README.md CHANGED
@@ -18,8 +18,8 @@ should probably proofread and complete it, then remove this comment. -->
18
 
19
  This model is a fine-tuned version of [google-bert/bert-base-uncased](https://huggingface.co/google-bert/bert-base-uncased) on the None dataset.
20
  It achieves the following results on the evaluation set:
21
- - Loss: 0.0001
22
- - Accuracy: 1.0
23
 
24
  ## Model description
25
 
@@ -50,16 +50,16 @@ The following hyperparameters were used during training:
50
 
51
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
52
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
53
- | 0.1348 | 1.0 | 350 | 0.0084 | 0.998 |
54
- | 0.0214 | 2.0 | 700 | 0.0020 | 1.0 |
55
- | 0.017 | 3.0 | 1050 | 0.0094 | 0.997 |
56
- | 0.0117 | 4.0 | 1400 | 0.0005 | 1.0 |
57
- | 0.0127 | 5.0 | 1750 | 0.0003 | 1.0 |
58
- | 0.0099 | 6.0 | 2100 | 0.0004 | 1.0 |
59
- | 0.0079 | 7.0 | 2450 | 0.0002 | 1.0 |
60
- | 0.0052 | 8.0 | 2800 | 0.0002 | 1.0 |
61
- | 0.0047 | 9.0 | 3150 | 0.0001 | 1.0 |
62
- | 0.006 | 10.0 | 3500 | 0.0001 | 1.0 |
63
 
64
 
65
  ### Framework versions
 
18
 
19
  This model is a fine-tuned version of [google-bert/bert-base-uncased](https://huggingface.co/google-bert/bert-base-uncased) on the None dataset.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 0.0034
22
+ - Accuracy: 0.999
23
 
24
  ## Model description
25
 
 
50
 
51
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
52
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
53
+ | 0.1228 | 1.0 | 657 | 0.0123 | 0.997 |
54
+ | 0.0172 | 2.0 | 1314 | 0.0088 | 0.999 |
55
+ | 0.0127 | 3.0 | 1971 | 0.0051 | 0.999 |
56
+ | 0.0068 | 4.0 | 2628 | 0.0037 | 0.999 |
57
+ | 0.0073 | 5.0 | 3285 | 0.0048 | 0.999 |
58
+ | 0.0088 | 6.0 | 3942 | 0.0046 | 0.999 |
59
+ | 0.0054 | 7.0 | 4599 | 0.0021 | 0.999 |
60
+ | 0.0048 | 8.0 | 5256 | 0.0037 | 0.999 |
61
+ | 0.0055 | 9.0 | 5913 | 0.0037 | 0.999 |
62
+ | 0.0048 | 10.0 | 6570 | 0.0034 | 0.999 |
63
 
64
 
65
  ### Framework versions
config.json CHANGED
@@ -11,13 +11,15 @@
11
  "hidden_size": 768,
12
  "id2label": {
13
  "0": "gardening",
14
- "1": "furniture"
 
15
  },
16
  "initializer_range": 0.02,
17
  "intermediate_size": 3072,
18
  "label2id": {
19
  "furniture": 1,
20
- "gardening": 0
 
21
  },
22
  "layer_norm_eps": 1e-12,
23
  "max_position_embeddings": 512,
 
11
  "hidden_size": 768,
12
  "id2label": {
13
  "0": "gardening",
14
+ "1": "furniture",
15
+ "2": "healthcare"
16
  },
17
  "initializer_range": 0.02,
18
  "intermediate_size": 3072,
19
  "label2id": {
20
  "furniture": 1,
21
+ "gardening": 0,
22
+ "healthcare": 2
23
  },
24
  "layer_norm_eps": 1e-12,
25
  "max_position_embeddings": 512,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:157625eab85d9bae5ffd73be5a309e9ae4345de2c2a3fbacd493c3bbe34d13ae
3
- size 437958648
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9d8b322127748ad2787ac88c001f4957e2e52f6704dcfd415516ea7eb33e490c
3
+ size 437961724
runs/Oct29_14-25-13_e7ef80ecc11b/events.out.tfevents.1761747921.e7ef80ecc11b.202.11 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8fb274e5ba1d8c79349f74b43b2051ef81f21c7fd0eada965e1976969eba477f
3
+ size 10899
runs/Oct29_14-30-53_e7ef80ecc11b/events.out.tfevents.1761748257.e7ef80ecc11b.202.12 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:418baad811e10f128859967d21d623f5235c6d44c729df7865c32bc29a0f6d92
3
+ size 10899
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:cc746027e3a90ef1974ddb63415f0892d34cd5381ed995f43be94bf9bb160524
3
  size 5841
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7be4934874cf24615879c22e0f18b6c2336ed13950d029b5dca754833a4033e6
3
  size 5841