Hugging Face's logo

bendjafer
/
yelp_review_classifier

TensorBoard
Safetensors
bert
Generated from Trainer
Model card Files
xet
Metrics Community
bendjafer commited on
Commit
d67c162
·
verified ·
1 Parent(s): d7f0db2

End of training

Browse files
Files changed (19) hide show
  1. README.md +61 -0
  2. config.json +53 -0
  3. model.safetensors +3 -0
  4. runs/Dec05_21-54-15_a743865a9bc3/events.out.tfevents.1764971656.a743865a9bc3.245.0 +3 -0
  5. runs/Dec05_21-58-38_a743865a9bc3/events.out.tfevents.1764971919.a743865a9bc3.245.1 +3 -0
  6. runs/Dec05_22-03-40_a743865a9bc3/events.out.tfevents.1764972220.a743865a9bc3.245.2 +3 -0
  7. runs/Dec05_23-03-34_a743865a9bc3/events.out.tfevents.1764975834.a743865a9bc3.245.3 +3 -0
  8. runs/Dec05_23-03-34_a743865a9bc3/events.out.tfevents.1764976159.a743865a9bc3.245.4 +3 -0
  9. runs/Dec05_23-42-26_a743865a9bc3/events.out.tfevents.1764978158.a743865a9bc3.48260.0 +3 -0
  10. runs/Dec06_00-16-03_a743865a9bc3/events.out.tfevents.1764980172.a743865a9bc3.52975.0 +3 -0
  11. runs/Dec06_00-18-56_a743865a9bc3/events.out.tfevents.1764980339.a743865a9bc3.52975.1 +3 -0
  12. runs/Dec06_00-22-37_a743865a9bc3/events.out.tfevents.1764980564.a743865a9bc3.57385.0 +3 -0
  13. runs/Dec06_00-22-37_a743865a9bc3/events.out.tfevents.1764980627.a743865a9bc3.57385.1 +3 -0
  14. runs/Dec06_00-25-58_a743865a9bc3/events.out.tfevents.1764981395.a743865a9bc3.58600.0 +3 -0
  15. runs/Dec06_00-43-24_a743865a9bc3/events.out.tfevents.1764981809.a743865a9bc3.58600.1 +3 -0
  16. runs/Dec06_00-51-31_a743865a9bc3/events.out.tfevents.1764982299.a743865a9bc3.63453.0 +3 -0
  17. runs/Dec06_01-04-27_a743865a9bc3/events.out.tfevents.1764983076.a743865a9bc3.67501.0 +3 -0
  18. runs/Dec06_01-06-31_a743865a9bc3/events.out.tfevents.1764983195.a743865a9bc3.67501.1 +3 -0
  19. training_args.bin +3 -0
README.md ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: google-bert/bert-base-cased
4
+ tags:
5
+ - generated_from_trainer
6
+ metrics:
7
+ - accuracy
8
+ model-index:
9
+ - name: yelp_review_classifier
10
+ results: []
11
+ ---
12
+
13
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
14
+ should probably proofread and complete it, then remove this comment. -->
15
+
16
+ # yelp_review_classifier
17
+
18
+ This model is a fine-tuned version of [google-bert/bert-base-cased](https://huggingface.co/google-bert/bert-base-cased) on the None dataset.
19
+ It achieves the following results on the evaluation set:
20
+ - Loss: 2.3225
21
+ - Accuracy: 0.0
22
+
23
+ ## Model description
24
+
25
+ More information needed
26
+
27
+ ## Intended uses & limitations
28
+
29
+ More information needed
30
+
31
+ ## Training and evaluation data
32
+
33
+ More information needed
34
+
35
+ ## Training procedure
36
+
37
+ ### Training hyperparameters
38
+
39
+ The following hyperparameters were used during training:
40
+ - learning_rate: 5e-05
41
+ - train_batch_size: 8
42
+ - eval_batch_size: 8
43
+ - seed: 42
44
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
+ - lr_scheduler_type: linear
46
+ - num_epochs: 2
47
+
48
+ ### Training results
49
+
50
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy |
51
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|
52
+ | No log | 1.0 | 1 | 2.3727 | 0.0 |
53
+ | No log | 2.0 | 2 | 2.3225 | 0.0 |
54
+
55
+
56
+ ### Framework versions
57
+
58
+ - Transformers 4.40.1
59
+ - Pytorch 2.4.0+cu121
60
+ - Datasets 2.19.1
61
+ - Tokenizers 0.19.1
config.json ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "google-bert/bert-base-cased",
3
+ "architectures": [
4
+ "BertForSequenceClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "classifier_dropout": null,
8
+ "gradient_checkpointing": false,
9
+ "hidden_act": "gelu",
10
+ "hidden_dropout_prob": 0.1,
11
+ "hidden_size": 768,
12
+ "id2label": {
13
+ "0": "B-Indicator",
14
+ "1": "B-Malware",
15
+ "2": "B-Organization",
16
+ "3": "B-System",
17
+ "4": "B-Vulnerability",
18
+ "5": "I-Indicator",
19
+ "6": "I-Malware",
20
+ "7": "I-Organization",
21
+ "8": "I-System",
22
+ "9": "I-Vulnerability",
23
+ "10": "O"
24
+ },
25
+ "initializer_range": 0.02,
26
+ "intermediate_size": 3072,
27
+ "label2id": {
28
+ "B-Indicator": 0,
29
+ "B-Malware": 1,
30
+ "B-Organization": 2,
31
+ "B-System": 3,
32
+ "B-Vulnerability": 4,
33
+ "I-Indicator": 5,
34
+ "I-Malware": 6,
35
+ "I-Organization": 7,
36
+ "I-System": 8,
37
+ "I-Vulnerability": 9,
38
+ "O": 10
39
+ },
40
+ "layer_norm_eps": 1e-12,
41
+ "max_position_embeddings": 512,
42
+ "model_type": "bert",
43
+ "num_attention_heads": 12,
44
+ "num_hidden_layers": 12,
45
+ "pad_token_id": 0,
46
+ "position_embedding_type": "absolute",
47
+ "problem_type": "single_label_classification",
48
+ "torch_dtype": "float32",
49
+ "transformers_version": "4.40.1",
50
+ "type_vocab_size": 2,
51
+ "use_cache": true,
52
+ "vocab_size": 28996
53
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:df8d6e2cc2fba59236ce46ce6351c7e1f0b2bdacf8198f375228323997eb274c
3
+ size 433298452
runs/Dec05_21-54-15_a743865a9bc3/events.out.tfevents.1764971656.a743865a9bc3.245.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:68ff4f195c59cbaab2e7990cfbbd58688d30495d9e7c61fc43c9e6dccd05bc97
3
+ size 5493
runs/Dec05_21-58-38_a743865a9bc3/events.out.tfevents.1764971919.a743865a9bc3.245.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:627622d88d652edefbdc8472566093add5a5cabafec5c9f767dabafb05ef8417
3
+ size 5493
runs/Dec05_22-03-40_a743865a9bc3/events.out.tfevents.1764972220.a743865a9bc3.245.2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eca5f3b48c27b706068721af1795773216404dfd2cbcb79e1832e6520e764887
3
+ size 5567
runs/Dec05_23-03-34_a743865a9bc3/events.out.tfevents.1764975834.a743865a9bc3.245.3 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:97ff1c1ffb7709b27c956073613fb51559376cb70cf1c1a72d894e8ca467c2bc
3
+ size 5260
runs/Dec05_23-03-34_a743865a9bc3/events.out.tfevents.1764976159.a743865a9bc3.245.4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4aa7712f97b684c5d9dacbc6f32a80c6992bb4f5cd259ecff9afdfb030ad32d9
3
+ size 4184
runs/Dec05_23-42-26_a743865a9bc3/events.out.tfevents.1764978158.a743865a9bc3.48260.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:99f125e918fbd994a0cd140688c4f6e3e229c6d62e79ea34fb7a2fb4ccc017b5
3
+ size 4933
runs/Dec06_00-16-03_a743865a9bc3/events.out.tfevents.1764980172.a743865a9bc3.52975.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ef7feb6e5cd452f7c3e39b88927f8fb598de151f9eb8b28ea122911b1c6e94e9
3
+ size 4933
runs/Dec06_00-18-56_a743865a9bc3/events.out.tfevents.1764980339.a743865a9bc3.52975.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2355c42b017dca980d6a80d6a5c0cc4cd444d938ff2c53b7f079c85f419a2f98
3
+ size 4184
runs/Dec06_00-22-37_a743865a9bc3/events.out.tfevents.1764980564.a743865a9bc3.57385.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:84d23aa62c4b29dc3e17777bcac7ef90bb0d7d17c5a4ff23b3c085a1198bd2c1
3
+ size 4184
runs/Dec06_00-22-37_a743865a9bc3/events.out.tfevents.1764980627.a743865a9bc3.57385.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7f008b8d1516ae613718801ee87abd3fff283570ad34c23da882dbd02433f62e
3
+ size 4184
runs/Dec06_00-25-58_a743865a9bc3/events.out.tfevents.1764981395.a743865a9bc3.58600.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cc729ae16e8addffe23767473b845155fc8b649efc2fc6bc781a00402f1edd7f
3
+ size 5239
runs/Dec06_00-43-24_a743865a9bc3/events.out.tfevents.1764981809.a743865a9bc3.58600.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d039b2738869f4c4fb2b2d350c2bd9a32f9f10721f140e27e82d3bfea5f572e6
3
+ size 4184
runs/Dec06_00-51-31_a743865a9bc3/events.out.tfevents.1764982299.a743865a9bc3.63453.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:43211a762741420b09a7c871c0299796fba34c8a4ceea3bee76986473d5ea4eb
3
+ size 5237
runs/Dec06_01-04-27_a743865a9bc3/events.out.tfevents.1764983076.a743865a9bc3.67501.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aca8f06a3c3424b0f0e45dbd9933495bf9a52a298f58d53ee9aeaa61bd17e7b0
3
+ size 5237
runs/Dec06_01-06-31_a743865a9bc3/events.out.tfevents.1764983195.a743865a9bc3.67501.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9493ae63f5da39d6ba56c55fbc1f0f9d106a87ae3104455c3a797a1478680d8e
3
+ size 6219
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a6b710b37eeae1481810766dd014ff07b9a83d10daf53133e1c90814fabd50c8
3
+ size 4984