Upload DistilBertForSequenceClassification

config.json

@@ -10,7 +10,15 @@
   "dtype": "float32",
   "eos_token_id": null,
   "hidden_dim": 3072,
+  "id2label": {
+    "0": "NOT_CHECKIN",
+    "1": "CHECKIN"
+  },
   "initializer_range": 0.02,
+  "label2id": {
+    "CHECKIN": 1,
+    "NOT_CHECKIN": 0
+  },
   "max_position_embeddings": 512,
   "model_type": "distilbert",
   "n_heads": 12,
@@ -23,6 +31,5 @@
   "tie_weights_": true,
   "tie_word_embeddings": true,
   "transformers_version": "5.0.0",
-  "use_cache": false,
   "vocab_size": 30522
 }