Ysfxjo commited on
Commit
b5b573b
·
verified ·
1 Parent(s): ea3ce48

Set label2id/id2label to match training schema

Browse files
Files changed (2) hide show
  1. config.json +13 -8
  2. model.safetensors +2 -2
config.json CHANGED
@@ -1,25 +1,30 @@
1
  {
 
2
  "architectures": [
3
  "BertForSequenceClassification"
4
  ],
5
  "attention_probs_dropout_prob": 0.1,
 
6
  "classifier_dropout": null,
7
  "directionality": "bidi",
 
 
8
  "gradient_checkpointing": false,
9
  "hidden_act": "gelu",
10
  "hidden_dropout_prob": 0.1,
11
  "hidden_size": 768,
12
  "id2label": {
13
- "0": "LABEL_0",
14
- "1": "LABEL_1",
15
- "2": "LABEL_2"
16
  },
17
  "initializer_range": 0.02,
18
  "intermediate_size": 3072,
 
19
  "label2id": {
20
- "LABEL_0": 0,
21
- "LABEL_1": 1,
22
- "LABEL_2": 2
23
  },
24
  "layer_norm_eps": 1e-12,
25
  "max_position_embeddings": 512,
@@ -34,8 +39,8 @@
34
  "pooler_type": "first_token_transform",
35
  "position_embedding_type": "absolute",
36
  "problem_type": "single_label_classification",
37
- "torch_dtype": "float32",
38
- "transformers_version": "4.53.3",
39
  "type_vocab_size": 2,
40
  "use_cache": true,
41
  "vocab_size": 100000
 
1
  {
2
+ "add_cross_attention": false,
3
  "architectures": [
4
  "BertForSequenceClassification"
5
  ],
6
  "attention_probs_dropout_prob": 0.1,
7
+ "bos_token_id": null,
8
  "classifier_dropout": null,
9
  "directionality": "bidi",
10
+ "dtype": "float32",
11
+ "eos_token_id": null,
12
  "gradient_checkpointing": false,
13
  "hidden_act": "gelu",
14
  "hidden_dropout_prob": 0.1,
15
  "hidden_size": 768,
16
  "id2label": {
17
+ "0": "NEG",
18
+ "1": "NEU",
19
+ "2": "POS"
20
  },
21
  "initializer_range": 0.02,
22
  "intermediate_size": 3072,
23
+ "is_decoder": false,
24
  "label2id": {
25
+ "NEG": 0,
26
+ "NEU": 1,
27
+ "POS": 2
28
  },
29
  "layer_norm_eps": 1e-12,
30
  "max_position_embeddings": 512,
 
39
  "pooler_type": "first_token_transform",
40
  "position_embedding_type": "absolute",
41
  "problem_type": "single_label_classification",
42
+ "tie_word_embeddings": true,
43
+ "transformers_version": "5.4.0",
44
  "type_vocab_size": 2,
45
  "use_cache": true,
46
  "vocab_size": 100000
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ad2d1db7c58f97ffe5bedcbe2ab605c50a559385604354d7e62c086347fbf2f9
3
- size 651398148
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:adc34e007816de05ceee6a4c796cee04b33d191de8dc3f9d2bfe78150d1dbc3c
3
+ size 651398124