Upload BertForSequenceClassification

README.md

@@ -6,27 +6,10 @@ tags:
 - bert
 - transformers
 - huggingface
-
 license: mit
 language: en
 datasets:
-- mrm8488/NLP-Twitter-disaster  # HF dataset id used instead of Kaggle
-model-index:
-- name: AI4SDG-BERT
-  results:
-    - task:
-        name: Text Classification
-        type: text-classification
-      metrics:
-        - name: Accuracy
-          type: accuracy
-          value: 0.92
-        - name: F1
-          type: f1
-          value: 0.91
-        - name: AUC
-          type: auc
-          value: 0.94
+- mrm8488/NLP-Twitter-disaster
 ---
 
 # AI4SDG-BERT Model Card

config.json

@@ -8,8 +8,16 @@
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.1,
   "hidden_size": 768,
+  "id2label": {
+    "0": "not_disaster",
+    "1": "disaster"
+  },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
+  "label2id": {
+    "disaster": 1,
+    "not_disaster": 0
+  },
   "layer_norm_eps": 1e-12,
   "max_position_embeddings": 512,
   "model_type": "bert",