philschmid HF staff commited on
Commit
72e8fa3
1 Parent(s): 4921590

add missing fiels to correctly load without model class

Browse files
Files changed (1) hide show
  1. config.json +15 -8
config.json CHANGED
@@ -1,26 +1,33 @@
1
  {
 
2
  "architectures": [
3
  "BertForSequenceClassification"
4
  ],
 
 
 
 
 
5
  "id2label": {
6
  "0": "Neutral",
7
  "1": "Positive",
8
  "2": "Negative"
9
  },
 
 
10
  "label2id": {
11
- "Positive": 1,
12
  "Negative": 2,
13
- "Neutral": 0
 
14
  },
15
- "attention_probs_dropout_prob": 0.1,
16
- "hidden_act": "gelu",
17
- "hidden_dropout_prob": 0.1,
18
- "hidden_size": 768,
19
- "initializer_range": 0.02,
20
- "intermediate_size": 3072,
21
  "max_position_embeddings": 512,
 
22
  "num_attention_heads": 12,
23
  "num_hidden_layers": 12,
 
 
 
24
  "type_vocab_size": 2,
25
  "vocab_size": 30873
26
  }
 
1
  {
2
+ "_name_or_path": "yiyanghkust/finbert-tone",
3
  "architectures": [
4
  "BertForSequenceClassification"
5
  ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "classifier_dropout": null,
8
+ "hidden_act": "gelu",
9
+ "hidden_dropout_prob": 0.1,
10
+ "hidden_size": 768,
11
  "id2label": {
12
  "0": "Neutral",
13
  "1": "Positive",
14
  "2": "Negative"
15
  },
16
+ "initializer_range": 0.02,
17
+ "intermediate_size": 3072,
18
  "label2id": {
 
19
  "Negative": 2,
20
+ "Neutral": 0,
21
+ "Positive": 1
22
  },
23
+ "layer_norm_eps": 1e-12,
 
 
 
 
 
24
  "max_position_embeddings": 512,
25
+ "model_type": "bert",
26
  "num_attention_heads": 12,
27
  "num_hidden_layers": 12,
28
+ "pad_token_id": 0,
29
+ "position_embedding_type": "absolute",
30
+ "torch_dtype": "float32",
31
  "type_vocab_size": 2,
32
  "vocab_size": 30873
33
  }