elozano
/

bert-base-cased-news-category

Text Classification

Inference Endpoints

Model card Files Files and versions Community

elozano commited on Mar 1, 2022

Commit

fbdaa11

•

1 Parent(s): dcc3965

Update config.json

Files changed (1) hide show

config.json +14 -14

config.json CHANGED Viewed

@@ -10,24 +10,24 @@
   "hidden_dropout_prob": 0.1,
   "hidden_size": 768,
   "id2label": {
-    "0": "automobile",
-    "1": "entertainment",
-    "2": "politics",
-    "3": "science",
-    "4": "sports",
-    "5": "technology",
-    "6": "world"
   },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "automobile": 0,
-    "entertainment": 1,
-    "politics": 2,
-    "science": 3,
-    "sports": 4,
-    "technology": 5,
-    "world": 6
   },
   "layer_norm_eps": 1e-12,
   "max_position_embeddings": 512,

   "hidden_dropout_prob": 0.1,
   "hidden_size": 768,
   "id2label": {
+    "0": "Automobile",
+    "1": "Entertainment",
+    "2": "Politics",
+    "3": "Science",
+    "4": "Sports",
+    "5": "Technology",
+    "6": "World"
   },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
+    "Automobile": 0,
+    "Entertainment": 1,
+    "Politics": 2,
+    "Science": 3,
+    "Sports": 4,
+    "Technology": 5,
+    "World": 6
   },
   "layer_norm_eps": 1e-12,
   "max_position_embeddings": 512,