MihanTilk
/

News_Classifier

Model card Files Files and versions

Update config.json

#1

by TharushiPerera - opened Mar 30, 2025

base: refs/heads/main

←

from: refs/pr/1

Discussion Files changed

Files changed (1) hide show

config.json +2 -0

config.json CHANGED Viewed

@@ -13,6 +13,8 @@
     "2": "Political_gossip",
     "3": "Sports",
     "4": "World_news"
   },
   "initializer_range": 0.02,
   "label2id": {

     "2": "Political_gossip",
     "3": "Sports",
     "4": "World_news"
+    "auto_map": {
+    "AutoModelForSequenceClassification": "modeling_distilbert.DistilBertForSequenceClassification"
   },
   "initializer_range": 0.02,
   "label2id": {