Simran KC commited on
Commit
cadffb6
·
1 Parent(s): 9c77776

Add label to id mapping in config

Browse files
Files changed (1) hide show
  1. config.json +5 -5
config.json CHANGED
@@ -20,11 +20,11 @@
20
  "initializer_range": 0.02,
21
  "intermediate_size": 3072,
22
  "label2id": {
23
- "LABEL_0": 0,
24
- "LABEL_1": 1,
25
- "LABEL_2": 2,
26
- "LABEL_3": 3,
27
- "LABEL_4": 4
28
  },
29
  "layer_norm_eps": 1e-05,
30
  "max_position_embeddings": 514,
 
20
  "initializer_range": 0.02,
21
  "intermediate_size": 3072,
22
  "label2id": {
23
+ "Astrophysics": 0,
24
+ "Biological and Physical Sciences": 1,
25
+ "Earth Science": 2,
26
+ "Heliophysics": 3,
27
+ "Planetary Science": 4
28
  },
29
  "layer_norm_eps": 1e-05,
30
  "max_position_embeddings": 514,