nouhadziri commited on
Commit
ed7f709
1 Parent(s): 3db57d2

label names added

Browse files
Files changed (1) hide show
  1. config.json +8 -0
config.json CHANGED
@@ -10,8 +10,16 @@
10
  "hidden_act": "gelu",
11
  "hidden_dropout_prob": 0.1,
12
  "hidden_size": 1024,
 
 
 
 
13
  "initializer_range": 0.02,
14
  "intermediate_size": 4096,
 
 
 
 
15
  "layer_norm_eps": 1e-05,
16
  "max_position_embeddings": 514,
17
  "model_type": "roberta",
 
10
  "hidden_act": "gelu",
11
  "hidden_dropout_prob": 0.1,
12
  "hidden_size": 1024,
13
+ "id2label": {
14
+ "0": "Entailment",
15
+ "1": "Hallucination"
16
+ },
17
  "initializer_range": 0.02,
18
  "intermediate_size": 4096,
19
+ "label2id": {
20
+ "Entailment": "0",
21
+ "Hallucination": "1"
22
+ },
23
  "layer_norm_eps": 1e-05,
24
  "max_position_embeddings": 514,
25
  "model_type": "roberta",