flax-community
/

roberta-swahili-news-classification

Text Classification Transformers PyTorch JAX TensorBoard Safetensors

Swahili roberta Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

fgaim commited on Jul 13, 2021

Commit

cea6e9e

•

1 Parent(s): f447518

Update config

Files changed (1) hide show

config.json +10 -10

config.json CHANGED Viewed

@@ -12,22 +12,22 @@
   "hidden_dropout_prob": 0.1,
   "hidden_size": 768,
   "id2label": {
-    "0": "uchumi",
-    "1":"kitaifa",
-    "2":"michezo",
-    "3":"kimataifa",
-    "4":"burudani",
-    "5":"afya"
   },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "uchumi": 0,
     "kitaifa": 1,
     "michezo": 2,
-    "kimataifa": 3,
-    "burudani": 4,
-    "afya": 5
   },
   "layer_norm_eps": 1e-05,
   "max_position_embeddings": 514,

   "hidden_dropout_prob": 0.1,
   "hidden_size": 768,
   "id2label": {
+    "0": "uchumi-economy",
+    "1": "kitaifa-national",
+    "2": "michezo-sports",
+    "3": "kimataifa-international",
+    "4": "burudani-entertainment",
+    "5": "afya-health"
   },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
+    "afya": 5,
+    "burudani": 4,
+    "kimataifa": 3,
     "kitaifa": 1,
     "michezo": 2,
+    "uchumi": 0
   },
   "layer_norm_eps": 1e-05,
   "max_position_embeddings": 514,