software-si commited on
Commit
57a2a2c
·
verified ·
1 Parent(s): 0b8b560

Upload BertForSequenceClassification

Browse files
Files changed (2) hide show
  1. config.json +16 -14
  2. model.safetensors +1 -1
config.json CHANGED
@@ -5,28 +5,29 @@
5
  "attention_probs_dropout_prob": 0.1,
6
  "classifier_dropout": null,
7
  "directionality": "bidi",
 
8
  "hidden_act": "gelu",
9
  "hidden_dropout_prob": 0.1,
10
  "hidden_size": 768,
11
  "id2label": {
12
- "0": "english",
13
- "1": "french",
14
- "2": "german",
15
- "3": "italian",
16
- "4": "notallowed",
17
- "5": "other",
18
- "6": "spanish"
19
  },
20
  "initializer_range": 0.02,
21
  "intermediate_size": 3072,
22
  "label2id": {
23
- "english": 0,
24
- "french": 1,
25
- "german": 2,
26
- "italian": 3,
27
- "notallowed": 4,
28
- "other": 5,
29
- "spanish": 6
30
  },
31
  "layer_norm_eps": 1e-12,
32
  "max_position_embeddings": 512,
@@ -40,6 +41,7 @@
40
  "pooler_size_per_head": 128,
41
  "pooler_type": "first_token_transform",
42
  "position_embedding_type": "absolute",
 
43
  "transformers_version": "4.56.2",
44
  "type_vocab_size": 2,
45
  "use_cache": true,
 
5
  "attention_probs_dropout_prob": 0.1,
6
  "classifier_dropout": null,
7
  "directionality": "bidi",
8
+ "dtype": "float32",
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
+ "3": "LABEL_3",
17
+ "4": "LABEL_4",
18
+ "5": "LABEL_5",
19
+ "6": "LABEL_6"
20
  },
21
  "initializer_range": 0.02,
22
  "intermediate_size": 3072,
23
  "label2id": {
24
+ "LABEL_0": 0,
25
+ "LABEL_1": 1,
26
+ "LABEL_2": 2,
27
+ "LABEL_3": 3,
28
+ "LABEL_4": 4,
29
+ "LABEL_5": 5,
30
+ "LABEL_6": 6
31
  },
32
  "layer_norm_eps": 1e-12,
33
  "max_position_embeddings": 512,
 
41
  "pooler_size_per_head": 128,
42
  "pooler_type": "first_token_transform",
43
  "position_embedding_type": "absolute",
44
+ "problem_type": "multi_label_classification",
45
  "transformers_version": "4.56.2",
46
  "type_vocab_size": 2,
47
  "use_cache": true,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:bdcb48cb648b2e143e57d751964df5ca6914cde56a452d468ded332c2172a049
3
  size 669470740
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cac7301b2fbba3328b63d6857084e9eb7731052154dcc28c7c28a553d8c666c1
3
  size 669470740