MihanTilk
/

News_Classifier

Model card Files Files and versions Community

Update config.json

#2

by TharushiPerera - opened 10 days ago

base: refs/heads/main

←

from: refs/pr/2

Discussion Files changed

Files changed (1) hide show

config.json +1 -3

config.json CHANGED Viewed

@@ -12,9 +12,7 @@
     "1": "Opinion",
     "2": "Political_gossip",
     "3": "Sports",
-    "4": "World_news"
-    "auto_map": {
-    "AutoModelForSequenceClassification": "modeling_distilbert.DistilBertForSequenceClassification"
   },
   "initializer_range": 0.02,
   "label2id": {

     "1": "Opinion",
     "2": "Political_gossip",
     "3": "Sports",
+      "4": "World_news"
   },
   "initializer_range": 0.02,
   "label2id": {