Hugging Face's logo

khalidrajan
/
distilbert-base-uncased_legal_ner_finetuned

TensorBoard
Safetensors
distilbert
Generated from Trainer
Model card Files
xet
 Metrics Community
khalidrajan commited on
Commit
11eacbd
·
verified ·
1 Parent(s): 4a23879

End of training

Browse files
Files changed (10) hide show
  1. README.md +87 -0
  2. config.json +46 -0
  3. model.safetensors +3 -0
  4. runs/Sep06_15-35-10_Khalids-MBP/events.out.tfevents.1725651311.Khalids-MBP.23384.8 +3 -0
  5. runs/Sep06_15-35-10_Khalids-MBP/events.out.tfevents.1725651887.Khalids-MBP.23384.9 +3 -0
  6. special_tokens_map.json +7 -0
  7. tokenizer.json +0 -0
  8. tokenizer_config.json +56 -0
  9. training_args.bin +3 -0
  10. vocab.txt +0 -0
README.md ADDED
@@ -0,0 +1,87 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: distilbert/distilbert-base-uncased
4
+ tags:
5
+ - generated_from_trainer
6
+ model-index:
7
+ - name: distilbert-base-uncased_legal_ner_finetuned
8
+ results: []
9
+ ---
10
+
11
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
12
+ should probably proofread and complete it, then remove this comment. -->
13
+
14
+ # distilbert-base-uncased_legal_ner_finetuned
15
+
16
+ This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on an unknown dataset.
17
+ It achieves the following results on the evaluation set:
18
+ - Loss: 0.2765
19
+ - Law Precision: 0.7983
20
+ - Law Recall: 0.8716
21
+ - Law F1: 0.8333
22
+ - Law Number: 109
23
+ - Violated by Precision: 0.7937
24
+ - Violated by Recall: 0.7042
25
+ - Violated by F1: 0.7463
26
+ - Violated by Number: 71
27
+ - Violated on Precision: 0.3934
28
+ - Violated on Recall: 0.3429
29
+ - Violated on F1: 0.3664
30
+ - Violated on Number: 70
31
+ - Violation Precision: 0.5657
32
+ - Violation Recall: 0.6588
33
+ - Violation F1: 0.6087
34
+ - Violation Number: 425
35
+ - Overall Precision: 0.6084
36
+ - Overall Recall: 0.6652
37
+ - Overall F1: 0.6355
38
+ - Overall Accuracy: 0.9409
39
+
40
+ ## Model description
41
+
42
+ More information needed
43
+
44
+ ## Intended uses & limitations
45
+
46
+ More information needed
47
+
48
+ ## Training and evaluation data
49
+
50
+ More information needed
51
+
52
+ ## Training procedure
53
+
54
+ ### Training hyperparameters
55
+
56
+ The following hyperparameters were used during training:
57
+ - learning_rate: 2e-05
58
+ - train_batch_size: 8
59
+ - eval_batch_size: 8
60
+ - seed: 42
61
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
62
+ - lr_scheduler_type: linear
63
+ - lr_scheduler_warmup_steps: 500
64
+ - num_epochs: 10
65
+
66
+ ### Training results
67
+
68
+ | Training Loss | Epoch | Step | Validation Loss | Law Precision | Law Recall | Law F1 | Law Number | Violated by Precision | Violated by Recall | Violated by F1 | Violated by Number | Violated on Precision | Violated on Recall | Violated on F1 | Violated on Number | Violation Precision | Violation Recall | Violation F1 | Violation Number | Overall Precision | Overall Recall | Overall F1 | Overall Accuracy |
69
+ |:-------------:|:-----:|:----:|:---------------:|:-------------:|:----------:|:------:|:----------:|:---------------------:|:------------------:|:--------------:|:------------------:|:---------------------:|:------------------:|:--------------:|:------------------:|:-------------------:|:----------------:|:------------:|:----------------:|:-----------------:|:--------------:|:----------:|:----------------:|
70
+ | No log | 1.0 | 85 | 1.1323 | 0.0 | 0.0 | 0.0 | 109 | 0.0 | 0.0 | 0.0 | 71 | 0.0 | 0.0 | 0.0 | 70 | 0.0 | 0.0 | 0.0 | 425 | 0.0 | 0.0 | 0.0 | 0.7656 |
71
+ | No log | 2.0 | 170 | 0.4593 | 0.0 | 0.0 | 0.0 | 109 | 0.0 | 0.0 | 0.0 | 71 | 0.0 | 0.0 | 0.0 | 70 | 0.1391 | 0.1741 | 0.1546 | 425 | 0.1391 | 0.1096 | 0.1226 | 0.8706 |
72
+ | No log | 3.0 | 255 | 0.3529 | 0.1923 | 0.0459 | 0.0741 | 109 | 0.0 | 0.0 | 0.0 | 71 | 0.0 | 0.0 | 0.0 | 70 | 0.2088 | 0.2 | 0.2043 | 425 | 0.2079 | 0.1333 | 0.1625 | 0.8943 |
73
+ | No log | 4.0 | 340 | 0.2708 | 0.1176 | 0.0734 | 0.0904 | 109 | 0.0 | 0.0 | 0.0 | 71 | 0.0 | 0.0 | 0.0 | 70 | 0.4321 | 0.4941 | 0.4610 | 425 | 0.3928 | 0.3230 | 0.3545 | 0.9134 |
74
+ | No log | 5.0 | 425 | 0.2579 | 0.8295 | 0.6697 | 0.7411 | 109 | 0.6667 | 0.3099 | 0.4231 | 71 | 0.3095 | 0.1857 | 0.2321 | 70 | 0.4197 | 0.4612 | 0.4395 | 425 | 0.4825 | 0.4504 | 0.4659 | 0.9153 |
75
+ | 0.5875 | 6.0 | 510 | 0.2516 | 0.8091 | 0.8165 | 0.8128 | 109 | 0.6 | 0.5070 | 0.5496 | 71 | 0.3542 | 0.2429 | 0.2881 | 70 | 0.5458 | 0.6588 | 0.5970 | 425 | 0.5773 | 0.6252 | 0.6003 | 0.9342 |
76
+ | 0.5875 | 7.0 | 595 | 0.2355 | 0.7946 | 0.8165 | 0.8054 | 109 | 0.7167 | 0.6056 | 0.6565 | 71 | 0.3438 | 0.3143 | 0.3284 | 70 | 0.5455 | 0.6353 | 0.5870 | 425 | 0.5800 | 0.6281 | 0.6031 | 0.9382 |
77
+ | 0.5875 | 8.0 | 680 | 0.2659 | 0.8246 | 0.8624 | 0.8430 | 109 | 0.7286 | 0.7183 | 0.7234 | 71 | 0.3243 | 0.3429 | 0.3333 | 70 | 0.5491 | 0.6706 | 0.6038 | 425 | 0.5843 | 0.6726 | 0.6253 | 0.9398 |
78
+ | 0.5875 | 9.0 | 765 | 0.2839 | 0.752 | 0.8624 | 0.8034 | 109 | 0.7391 | 0.7183 | 0.7286 | 71 | 0.3421 | 0.3714 | 0.3562 | 70 | 0.5524 | 0.6824 | 0.6105 | 425 | 0.5799 | 0.6830 | 0.6272 | 0.9394 |
79
+ | 0.5875 | 10.0 | 850 | 0.2765 | 0.7983 | 0.8716 | 0.8333 | 109 | 0.7937 | 0.7042 | 0.7463 | 71 | 0.3934 | 0.3429 | 0.3664 | 70 | 0.5657 | 0.6588 | 0.6087 | 425 | 0.6084 | 0.6652 | 0.6355 | 0.9409 |
80
+
81
+
82
+ ### Framework versions
83
+
84
+ - Transformers 4.44.0
85
+ - Pytorch 2.4.0
86
+ - Datasets 2.21.0
87
+ - Tokenizers 0.19.1
config.json ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "distilbert/distilbert-base-uncased",
3
+ "activation": "gelu",
4
+ "architectures": [
5
+ "DistilBertForTokenClassification"
6
+ ],
7
+ "attention_dropout": 0.1,
8
+ "dim": 768,
9
+ "dropout": 0.1,
10
+ "hidden_dim": 3072,
11
+ "id2label": {
12
+ "0": "O",
13
+ "1": "I-VIOLATED ON",
14
+ "2": "B-VIOLATED BY",
15
+ "3": "B-LAW",
16
+ "4": "I-LAW",
17
+ "5": "I-VIOLATED BY",
18
+ "6": "B-VIOLATED ON",
19
+ "7": "B-VIOLATION",
20
+ "8": "I-VIOLATION"
21
+ },
22
+ "initializer_range": 0.02,
23
+ "label2id": {
24
+ "B-LAW": 3,
25
+ "B-VIOLATED BY": 2,
26
+ "B-VIOLATED ON": 6,
27
+ "B-VIOLATION": 7,
28
+ "I-LAW": 4,
29
+ "I-VIOLATED BY": 5,
30
+ "I-VIOLATED ON": 1,
31
+ "I-VIOLATION": 8,
32
+ "O": 0
33
+ },
34
+ "max_position_embeddings": 512,
35
+ "model_type": "distilbert",
36
+ "n_heads": 12,
37
+ "n_layers": 6,
38
+ "pad_token_id": 0,
39
+ "qa_dropout": 0.1,
40
+ "seq_classif_dropout": 0.2,
41
+ "sinusoidal_pos_embds": false,
42
+ "tie_weights_": true,
43
+ "torch_dtype": "float32",
44
+ "transformers_version": "4.44.0",
45
+ "vocab_size": 30522
46
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c4efd6b3555119c773795514a6e5b197cfc3e0399a68990ae0297eea6eec14cf
3
+ size 265491548
runs/Sep06_15-35-10_Khalids-MBP/events.out.tfevents.1725651311.Khalids-MBP.23384.8 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:87318134d3cadcdab4b393acbb30fd5deeb06110dd37ba45a69ea6d7e2dc7741
3
+ size 20397
runs/Sep06_15-35-10_Khalids-MBP/events.out.tfevents.1725651887.Khalids-MBP.23384.9 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:47b56bbdd8a6613b45b0b13c6ea454da8f0edb17c4d39d2f94f46a63c6b41374
3
+ size 1540
special_tokens_map.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "cls_token": "[CLS]",
3
+ "mask_token": "[MASK]",
4
+ "pad_token": "[PAD]",
5
+ "sep_token": "[SEP]",
6
+ "unk_token": "[UNK]"
7
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": true,
3
+ "added_tokens_decoder": {
4
+ "0": {
5
+ "content": "[PAD]",
6
+ "lstrip": false,
7
+ "normalized": false,
8
+ "rstrip": false,
9
+ "single_word": false,
10
+ "special": true
11
+ },
12
+ "100": {
13
+ "content": "[UNK]",
14
+ "lstrip": false,
15
+ "normalized": false,
16
+ "rstrip": false,
17
+ "single_word": false,
18
+ "special": true
19
+ },
20
+ "101": {
21
+ "content": "[CLS]",
22
+ "lstrip": false,
23
+ "normalized": false,
24
+ "rstrip": false,
25
+ "single_word": false,
26
+ "special": true
27
+ },
28
+ "102": {
29
+ "content": "[SEP]",
30
+ "lstrip": false,
31
+ "normalized": false,
32
+ "rstrip": false,
33
+ "single_word": false,
34
+ "special": true
35
+ },
36
+ "103": {
37
+ "content": "[MASK]",
38
+ "lstrip": false,
39
+ "normalized": false,
40
+ "rstrip": false,
41
+ "single_word": false,
42
+ "special": true
43
+ }
44
+ },
45
+ "clean_up_tokenization_spaces": true,
46
+ "cls_token": "[CLS]",
47
+ "do_lower_case": true,
48
+ "mask_token": "[MASK]",
49
+ "model_max_length": 512,
50
+ "pad_token": "[PAD]",
51
+ "sep_token": "[SEP]",
52
+ "strip_accents": null,
53
+ "tokenize_chinese_chars": true,
54
+ "tokenizer_class": "DistilBertTokenizer",
55
+ "unk_token": "[UNK]"
56
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0024e2dcfa0aa081e3e6903e5219f617f5a3fbeb0eafc178d9df89a4aa97f1f0
3
+ size 5240
vocab.txt ADDED
The diff for this file is too large to render. See raw diff