crabz commited on
Commit
a84269f
·
1 Parent(s): a38afa9
README.md CHANGED
@@ -1,3 +1,95 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - generated_from_trainer
4
+ datasets:
5
+ - universal_dependencies
6
+ metrics:
7
+ - precision
8
+ - recall
9
+ - f1
10
+ - accuracy
11
+ model-index:
12
+ - name: distilbert-final-upos
13
+ results:
14
+ - task:
15
+ name: Token Classification
16
+ type: token-classification
17
+ dataset:
18
+ name: universal_dependencies sk_snk
19
+ type: universal_dependencies
20
+ args: sk_snk
21
+ metrics:
22
+ - name: Precision
23
+ type: precision
24
+ value: 0.9771104035797263
25
+ - name: Recall
26
+ type: recall
27
+ value: 0.9785418821096173
28
+ - name: F1
29
+ type: f1
30
+ value: 0.9778256189451022
31
+ - name: Accuracy
32
+ type: accuracy
33
+ value: 0.9800851200513933
34
+ ---
35
+
36
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
37
+ should probably proofread and complete it, then remove this comment. -->
38
+
39
+ # distilbert-final-upos
40
+
41
+ This model is a fine-tuned version of [../../../../distilbert-final](https://huggingface.co/../../../../distilbert-final) on the universal_dependencies sk_snk dataset.
42
+ It achieves the following results on the evaluation set:
43
+ - Loss: 0.1207
44
+ - Precision: 0.9771
45
+ - Recall: 0.9785
46
+ - F1: 0.9778
47
+ - Accuracy: 0.9801
48
+
49
+ ## Model description
50
+
51
+ More information needed
52
+
53
+ ## Intended uses & limitations
54
+
55
+ More information needed
56
+
57
+ ## Training and evaluation data
58
+
59
+ More information needed
60
+
61
+ ## Training procedure
62
+
63
+ ### Training hyperparameters
64
+
65
+ The following hyperparameters were used during training:
66
+ - learning_rate: 5e-05
67
+ - train_batch_size: 32
68
+ - eval_batch_size: 32
69
+ - seed: 42
70
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
71
+ - lr_scheduler_type: linear
72
+ - num_epochs: 10.0
73
+
74
+ ### Training results
75
+
76
+ | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
77
+ |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
78
+ | No log | 1.0 | 266 | 0.2168 | 0.9570 | 0.9554 | 0.9562 | 0.9610 |
79
+ | 0.3935 | 2.0 | 532 | 0.1416 | 0.9723 | 0.9736 | 0.9730 | 0.9740 |
80
+ | 0.3935 | 3.0 | 798 | 0.1236 | 0.9722 | 0.9735 | 0.9728 | 0.9747 |
81
+ | 0.0664 | 4.0 | 1064 | 0.1195 | 0.9722 | 0.9741 | 0.9732 | 0.9766 |
82
+ | 0.0664 | 5.0 | 1330 | 0.1160 | 0.9764 | 0.9772 | 0.9768 | 0.9789 |
83
+ | 0.0377 | 6.0 | 1596 | 0.1194 | 0.9763 | 0.9776 | 0.9770 | 0.9790 |
84
+ | 0.0377 | 7.0 | 1862 | 0.1188 | 0.9740 | 0.9755 | 0.9748 | 0.9777 |
85
+ | 0.024 | 8.0 | 2128 | 0.1188 | 0.9762 | 0.9777 | 0.9769 | 0.9793 |
86
+ | 0.024 | 9.0 | 2394 | 0.1207 | 0.9774 | 0.9789 | 0.9781 | 0.9802 |
87
+ | 0.0184 | 10.0 | 2660 | 0.1207 | 0.9771 | 0.9785 | 0.9778 | 0.9801 |
88
+
89
+
90
+ ### Framework versions
91
+
92
+ - Transformers 4.17.0.dev0
93
+ - Pytorch 1.10.0
94
+ - Datasets 1.16.1
95
+ - Tokenizers 0.11.0
config.json ADDED
@@ -0,0 +1,69 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "../../../../distilbert-final",
3
+ "architectures": [
4
+ "RobertaForTokenClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "bos_token_id": 0,
8
+ "classifier_dropout": null,
9
+ "eos_token_id": 2,
10
+ "finetuning_task": "ner",
11
+ "hidden_act": "gelu",
12
+ "hidden_dropout_prob": 0.1,
13
+ "hidden_size": 768,
14
+ "id2label": {
15
+ "0": "NOUN",
16
+ "1": "PUNCT",
17
+ "2": "ADP",
18
+ "3": "NUM",
19
+ "4": "SYM",
20
+ "5": "SCONJ",
21
+ "6": "ADJ",
22
+ "7": "PART",
23
+ "8": "DET",
24
+ "9": "CCONJ",
25
+ "10": "PROPN",
26
+ "11": "PRON",
27
+ "12": "X",
28
+ "13": "_",
29
+ "14": "ADV",
30
+ "15": "INTJ",
31
+ "16": "VERB",
32
+ "17": "AUX"
33
+ },
34
+ "initializer_range": 0.02,
35
+ "intermediate_size": 3072,
36
+ "label2id": {
37
+ "ADJ": 6,
38
+ "ADP": 2,
39
+ "ADV": 14,
40
+ "AUX": 17,
41
+ "CCONJ": 9,
42
+ "DET": 8,
43
+ "INTJ": 15,
44
+ "NOUN": 0,
45
+ "NUM": 3,
46
+ "PART": 7,
47
+ "PRON": 11,
48
+ "PROPN": 10,
49
+ "PUNCT": 1,
50
+ "SCONJ": 5,
51
+ "SYM": 4,
52
+ "VERB": 16,
53
+ "X": 12,
54
+ "_": 13
55
+ },
56
+ "layer_norm_eps": 1e-05,
57
+ "max_position_embeddings": 514,
58
+ "model_type": "roberta",
59
+ "num_attention_heads": 6,
60
+ "num_hidden_layers": 6,
61
+ "output_hidden_states": true,
62
+ "pad_token_id": 1,
63
+ "position_embedding_type": "absolute",
64
+ "torch_dtype": "float32",
65
+ "transformers_version": "4.17.0.dev0",
66
+ "type_vocab_size": 1,
67
+ "use_cache": true,
68
+ "vocab_size": 50264
69
+ }
merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e6784532c47008fbf125471cdc7f891415634ad80da4d6c52dbd879f41389802
3
+ size 326208945
special_tokens_map.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"bos_token": {"content": "<s>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true}, "eos_token": {"content": "</s>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true}, "unk_token": {"content": "<unk>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true}, "sep_token": {"content": "</s>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true}, "pad_token": {"content": "<pad>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true}, "cls_token": {"content": "<s>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true}, "mask_token": {"content": "<mask>", "single_word": false, "lstrip": true, "rstrip": false, "normalized": true}}
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"errors": "replace", "bos_token": {"content": "<s>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true, "__type": "AddedToken"}, "eos_token": {"content": "</s>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true, "__type": "AddedToken"}, "sep_token": {"content": "</s>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true, "__type": "AddedToken"}, "cls_token": {"content": "<s>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true, "__type": "AddedToken"}, "unk_token": {"content": "<unk>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true, "__type": "AddedToken"}, "pad_token": {"content": "<pad>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true, "__type": "AddedToken"}, "mask_token": {"content": "<mask>", "single_word": false, "lstrip": true, "rstrip": false, "normalized": true, "__type": "AddedToken"}, "add_prefix_space": true, "trim_offsets": true, "special_tokens_map_file": null, "model_max_length": 512, "name_or_path": "../../../../distilbert-final", "tokenizer_class": "RobertaTokenizer"}
vocab.json ADDED
The diff for this file is too large to render. See raw diff