serge-wilson
/

news_classification

Text Classification

Inference Endpoints

Model card Files Files and versions Community

serge-wilson commited on Aug 30, 2023

Commit

6ab7179

•

1 Parent(s): cd0b4ab

model deployment

Files changed (1) hide show

config.json +8 -8

config.json CHANGED Viewed

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

   "hidden_dropout_prob": 0.1,
   "hidden_size": 768,
   "id2label":{
+      "0": "technology",
+      "1": "sports",
+      "2": "science",
+      "3": "entertainment",
+      "4": "automobile",
+      "5": "world",
+      "6": "politics"
       },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
+    "technology":0,
     "sports": 1,
     "science": 2,
     "entertainment": 3,