StrangeSX commited on
Commit
a7fa50b
1 Parent(s): 84f9890

Update model config with label mappings

Browse files
Files changed (1) hide show
  1. config.json +59 -59
config.json CHANGED
@@ -1,59 +1,59 @@
1
- {
2
- "_name_or_path": "pythainlp/thainer-corpus-v2-base-model",
3
- "architectures": [
4
- "CamembertForTokenClassification"
5
- ],
6
- "attention_probs_dropout_prob": 0.1,
7
- "bos_token_id": 0,
8
- "classifier_dropout": null,
9
- "eos_token_id": 2,
10
- "gradient_checkpointing": false,
11
- "hidden_act": "gelu",
12
- "hidden_dropout_prob": 0.1,
13
- "hidden_size": 768,
14
- "id2label": {
15
- "0": "LABEL_0",
16
- "1": "LABEL_1",
17
- "2": "LABEL_2",
18
- "3": "LABEL_3",
19
- "4": "LABEL_4",
20
- "5": "LABEL_5",
21
- "6": "LABEL_6",
22
- "7": "LABEL_7",
23
- "8": "LABEL_8",
24
- "9": "LABEL_9",
25
- "10": "LABEL_10",
26
- "11": "LABEL_11",
27
- "12": "LABEL_12"
28
- },
29
- "initializer_range": 0.02,
30
- "intermediate_size": 3072,
31
- "label2id": {
32
- "LABEL_0": 0,
33
- "LABEL_1": 1,
34
- "LABEL_10": 10,
35
- "LABEL_11": 11,
36
- "LABEL_12": 12,
37
- "LABEL_2": 2,
38
- "LABEL_3": 3,
39
- "LABEL_4": 4,
40
- "LABEL_5": 5,
41
- "LABEL_6": 6,
42
- "LABEL_7": 7,
43
- "LABEL_8": 8,
44
- "LABEL_9": 9
45
- },
46
- "layer_norm_eps": 1e-12,
47
- "max_position_embeddings": 512,
48
- "model_type": "camembert",
49
- "num_attention_head": 12,
50
- "num_attention_heads": 12,
51
- "num_hidden_layers": 12,
52
- "pad_token_id": 1,
53
- "position_embedding_type": "absolute",
54
- "torch_dtype": "float32",
55
- "transformers_version": "4.46.3",
56
- "type_vocab_size": 1,
57
- "use_cache": true,
58
- "vocab_size": 25005
59
- }
 
1
+ {
2
+ "_name_or_path": "StrangeSX/NER-FT-Thai-Products-Group",
3
+ "architectures": [
4
+ "CamembertForTokenClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "bos_token_id": 0,
8
+ "classifier_dropout": null,
9
+ "eos_token_id": 2,
10
+ "gradient_checkpointing": false,
11
+ "hidden_act": "gelu",
12
+ "hidden_dropout_prob": 0.1,
13
+ "hidden_size": 768,
14
+ "id2label": {
15
+ "0": "O",
16
+ "1": "B-INGREDIENT",
17
+ "2": "I-INGREDIENT",
18
+ "3": "B-QUANTITY",
19
+ "4": "B-UNIT",
20
+ "5": "B-PACKAGE",
21
+ "6": "B-ATTRIBUTE",
22
+ "7": "I-ATTRIBUTE",
23
+ "8": "I-UNIT",
24
+ "9": "B-BRAND",
25
+ "10": "I-BRAND",
26
+ "11": "I-QUANTITY",
27
+ "12": "I-PACKAGE"
28
+ },
29
+ "initializer_range": 0.02,
30
+ "intermediate_size": 3072,
31
+ "label2id": {
32
+ "B-ATTRIBUTE": 6,
33
+ "B-BRAND": 9,
34
+ "B-INGREDIENT": 1,
35
+ "B-PACKAGE": 5,
36
+ "B-QUANTITY": 3,
37
+ "B-UNIT": 4,
38
+ "I-ATTRIBUTE": 7,
39
+ "I-BRAND": 10,
40
+ "I-INGREDIENT": 2,
41
+ "I-PACKAGE": 12,
42
+ "I-QUANTITY": 11,
43
+ "I-UNIT": 8,
44
+ "O": 0
45
+ },
46
+ "layer_norm_eps": 1e-12,
47
+ "max_position_embeddings": 512,
48
+ "model_type": "camembert",
49
+ "num_attention_head": 12,
50
+ "num_attention_heads": 12,
51
+ "num_hidden_layers": 12,
52
+ "pad_token_id": 1,
53
+ "position_embedding_type": "absolute",
54
+ "torch_dtype": "float32",
55
+ "transformers_version": "4.46.2",
56
+ "type_vocab_size": 1,
57
+ "use_cache": true,
58
+ "vocab_size": 25005
59
+ }