Anwaarma commited on
Commit
b67f464
1 Parent(s): c3d5ed3

Upload BertForSequenceClassification

Browse files
Files changed (2) hide show
  1. config.json +8 -1
  2. pytorch_model.bin +1 -1
config.json CHANGED
@@ -9,8 +9,16 @@
9
  "hidden_act": "gelu",
10
  "hidden_dropout_prob": 0.1,
11
  "hidden_size": 768,
 
 
 
 
12
  "initializer_range": 0.02,
13
  "intermediate_size": 3072,
 
 
 
 
14
  "layer_norm_eps": 1e-12,
15
  "max_position_embeddings": 512,
16
  "model_type": "bert",
@@ -19,7 +27,6 @@
19
  "output_past": true,
20
  "pad_token_id": 0,
21
  "position_embedding_type": "absolute",
22
- "problem_type": "single_label_classification",
23
  "torch_dtype": "float32",
24
  "transformers_version": "4.34.1",
25
  "type_vocab_size": 2,
 
9
  "hidden_act": "gelu",
10
  "hidden_dropout_prob": 0.1,
11
  "hidden_size": 768,
12
+ "id2label": {
13
+ "0": "Negative",
14
+ "1": "Positive"
15
+ },
16
  "initializer_range": 0.02,
17
  "intermediate_size": 3072,
18
+ "label2id": {
19
+ "Negative": 0,
20
+ "Positive": 1
21
+ },
22
  "layer_norm_eps": 1e-12,
23
  "max_position_embeddings": 512,
24
  "model_type": "bert",
 
27
  "output_past": true,
28
  "pad_token_id": 0,
29
  "position_embedding_type": "absolute",
 
30
  "torch_dtype": "float32",
31
  "transformers_version": "4.34.1",
32
  "type_vocab_size": 2,
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b7fb93aba66ed5dbdb163eb0b6320c45e71b263cbe0ce1c1f3c22f3c1a538c26
3
  size 442544366
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9289aa527222e69f118af29fe07e8b4672440dae93c06aa0c26ef81a9f125610
3
  size 442544366