xqewec
/

title_classifier

Text Classification

Model card Files Files and versions

xqewec commited on Jun 11, 2023

Commit

be12970

·

1 Parent(s): c89f22c

Update config.json

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -12,7 +12,7 @@
   "hidden_size": 768,
   "id2label": {
     "0": "LABEL_0",
-    "1": "LABEL_1",
     "2": "LABEL_2",
     "3": "LABEL_3",
     "4": "LABEL_4",
@@ -25,7 +25,7 @@
   "intermediate_size": 3072,
   "label2id": {
     "LABEL_0": 0,
-    "LABEL_1": 1,
     "LABEL_2": 2,
     "LABEL_3": 3,
     "LABEL_4": 4,

   "hidden_size": 768,
   "id2label": {
     "0": "LABEL_0",
+    "1": "cs.CL",
     "2": "LABEL_2",
     "3": "LABEL_3",
     "4": "LABEL_4",
   "intermediate_size": 3072,
   "label2id": {
     "LABEL_0": 0,
+    "cs.CL": 1,
     "LABEL_2": 2,
     "LABEL_3": 3,
     "LABEL_4": 4,