supreet-kore commited on
Commit
3537d83
1 Parent(s): 082cccc

Update config.json

Browse files

Adding labels for zeroshot classification pipeline

Files changed (1) hide show
  1. config.json +10 -0
config.json CHANGED
@@ -9,6 +9,16 @@
9
  "intermediate_size": 3072,
10
  "layer_norm_eps": 1e-12,
11
  "max_position_embeddings": 512,
 
 
 
 
 
 
 
 
 
 
12
  "model_type": "bert",
13
  "num_attention_heads": 12,
14
  "num_hidden_layers": 12,
 
9
  "intermediate_size": 3072,
10
  "layer_norm_eps": 1e-12,
11
  "max_position_embeddings": 512,
12
+ "id2label": {
13
+ "0": "entailment",
14
+ "1": "neutral",
15
+ "2": "contradiction"
16
+ },
17
+ "label2id": {
18
+ "contradiction": 2,
19
+ "entailment": 0,
20
+ "neutral": 1
21
+ },
22
  "model_type": "bert",
23
  "num_attention_heads": 12,
24
  "num_hidden_layers": 12,