wanderer2k1
/

BERT-query-classifier

Text Classification

Inference Endpoints

Model card Files Files and versions Community

wanderer2k1 commited on Aug 9, 2024

Commit

9d61fb2

·

verified ·

1 Parent(s): 8349e41

Upload BertForSequenceClassification

Files changed (2) hide show

config.json +3 -5
model.safetensors +2 -2

config.json CHANGED Viewed

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "ckpts/ckpt-epoch-1",
   "architectures": [
     "BertForSequenceClassification"
   ],
@@ -13,8 +13,7 @@
     "0": "LABEL_0",
     "1": "LABEL_1",
     "2": "LABEL_2",
-    "3": "LABEL_3",
-    "4": "LABEL_4"
   },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
@@ -22,8 +21,7 @@
     "LABEL_0": 0,
     "LABEL_1": 1,
     "LABEL_2": 2,
-    "LABEL_3": 3,
-    "LABEL_4": 4
   },
   "layer_norm_eps": 1e-12,
   "max_position_embeddings": 512,

 {
+  "_name_or_path": "ckpts/ckpt-epoch-2",
   "architectures": [
     "BertForSequenceClassification"
   ],
     "0": "LABEL_0",
     "1": "LABEL_1",
     "2": "LABEL_2",
+    "3": "LABEL_3"
   },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
     "LABEL_0": 0,
     "LABEL_1": 1,
     "LABEL_2": 2,
+    "LABEL_3": 3
   },
   "layer_norm_eps": 1e-12,
   "max_position_embeddings": 512,

model.safetensors CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:be86cb17e5cf5ee3fdd0a0f6ecc86aaa9275c7342e661712b09935aba3629955
-size 534668300

 version https://git-lfs.github.com/spec/v1
+oid sha256:73ecc4a6229f5334f491c5e17ee554e28d70a82b0d72bc0651282042d228bcd9
+size 534665224