Upload DistilBertForSequenceClassification

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "distilbert/distilbert-base-uncased",
+  "_name_or_path": "finetuned_TFPF_model",
   "activation": "gelu",
   "architectures": [
     "DistilBertForSequenceClassification"
@@ -9,71 +9,13 @@
   "dropout": 0.1,
   "hidden_dim": 3072,
   "id2label": {
-    "0": "REPORT INTEL LIST",
-    "1": "INTEGRATIONS",
-    "2": "REPORT GROUPS",
-    "3": "PROJECTS",
-    "4": "LIST THE USERS IN AN ORG",
-    "5": "LIST THE GROUPS IN AN ORG",
-    "6": "SECURITY POLICY",
-    "7": "DOCUMENTATION LINK",
-    "8": "ACTIVE PROJECTS",
-    "9": "SCANNERS LIST",
-    "10": "REPORT EXECUTIVE SUMMARY",
-    "11": "DOWNLOAD PDF",
-    "12": "WHOAMI",
-    "13": "APIKEYS",
-    "14": "NOTIFICATIONS",
-    "15": "API DOCS",
-    "16": "HEY",
-    "17": "ACCOUNT ACTIVITY",
-    "18": "VULNERABILITIES",
-    "19": "WEEKLY TREND",
-    "20": "DAILY TREND",
-    "21": "MONTHLY TREND",
-    "22": "TOTAL VIEW",
-    "23": "ACTIVE DEVELOPERS",
-    "24": "TPR",
-    "25": "FPR",
-    "26": "OWASP TOP 10",
-    "27": "PCI DSS",
-    "28": "NIST",
-    "29": "CWE TOP 25",
-    "30": "HIGH VULNERABILITIES"
+    "0": "FALSE_POSITIVE",
+    "1": "TRUE_POSITIVE"
   },
   "initializer_range": 0.02,
   "label2id": {
-    "ACCOUNT ACTIVITY": 17,
-    "ACTIVE DEVELOPERS": 23,
-    "ACTIVE PROJECTS": 8,
-    "API DOCS": 15,
-    "APIKEYS": 13,
-    "CWE TOP 25": 29,
-    "DAILY TREND": 20,
-    "DOCUMENTATION LINK": 7,
-    "DOWNLOAD PDF": 11,
-    "FPR": 25,
-    "HEY": 16,
-    "HIGH VULNERABILITIES": 30,
-    "INTEGRATIONS": 1,
-    "LIST THE GROUPS IN AN ORG": 5,
-    "LIST THE USERS IN AN ORG": 4,
-    "MONTHLY TREND": 21,
-    "NIST": 28,
-    "NOTIFICATIONS": 14,
-    "OWASP TOP 10": 26,
-    "PCI DSS": 27,
-    "PROJECTS": 3,
-    "REPORT EXECUTIVE SUMMARY": 10,
-    "REPORT GROUPS": 2,
-    "REPORT INTEL LIST": 0,
-    "SCANNERS LIST": 9,
-    "SECURITY POLICY": 6,
-    "TOTAL VIEW": 22,
-    "TPR": 24,
-    "VULNERABILITIES": 18,
-    "WEEKLY TREND": 19,
-    "WHOAMI": 12
+    "FALSE_POSITIVE": 0,
+    "TRUE_POSITIVE": 1
   },
   "max_position_embeddings": 512,
   "model_type": "distilbert",
@@ -86,6 +28,6 @@
   "sinusoidal_pos_embds": false,
   "tie_weights_": true,
   "torch_dtype": "float32",
-  "transformers_version": "4.47.1",
+  "transformers_version": "4.48.1",
   "vocab_size": 30522
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:563628d610aae262638b6dd8ec747490eba54807f7a6d737a9d21cc095f8c8c6
-size 267921780
+oid sha256:baaa67166c196c750aab309bd98b00c16397ed2b2adc8e21156120f9f9a906b8
+size 267832560