Upload v2 of URL classifier model

Training Metrics:
- Eval Loss: 0.033418405801057816
- Eval F1: 0.9921365145524408

README.md

@@ -9,10 +9,10 @@ tags:
 - domain-classification
 pipeline_tag: text-classification
 widget:
-- text: "https://acmewidgets.com"
-- text: "https://store.myshopify.com"
-- text: "https://example.wixsite.com/store"
-- text: "https://business.com"
+- text: https://acmewidgets.com
+- text: https://store.myshopify.com
+- text: https://example.wixsite.com/store
+- text: https://business.com
 datasets:
 - custom
 metrics:
@@ -20,25 +20,6 @@ metrics:
 - f1
 - precision
 - recall
-model-index:
-- name: urlbert-url-classifier
-  results:
-  - task:
-      type: text-classification
-      name: URL Classification
-    metrics:
-    - type: accuracy
-      value: 0.99
-      name: Test Accuracy
-    - type: f1
-      value: 0.99
-      name: Test F1 Score
-    - type: precision
-      value: 0.99
-      name: Test Precision
-    - type: recall
-      value: 0.99
-      name: Test Recall
 ---
 
 # URL Classifier

config.json

@@ -9,22 +9,14 @@
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.1,
   "hidden_size": 768,
-  "id2label": {
-    "0": "official_website",
-    "1": "platform"
-  },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
-  "label2id": {
-    "official_website": 0,
-    "platform": 1
-  },
   "layer_norm_eps": 1e-12,
   "max_position_embeddings": 512,
   "model_type": "bert",
   "num_attention_heads": 12,
   "num_hidden_layers": 12,
-  "pad_token_id": 1,
+  "pad_token_id": 0,
   "position_embedding_type": "absolute",
   "problem_type": "single_label_classification",
   "transformers_version": "4.57.0",

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e5fd6821d77beb5a96e00debad1a1b2ebe108f5a7ac0e87b07fbdc5e3160686a
+oid sha256:794e1e525e855085c30816b103992ff17a3d6b1363c30b897aca719ea81f0c74
 size 442499064