Aikaksh-Singh-Routela commited on
Commit
183bfb9
·
verified ·
1 Parent(s): 177cd53

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +62 -0
README.md ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: en
3
+ license: mit
4
+ tags:
5
+ - cybersecurity
6
+ - bert
7
+ - text-classification
8
+ - security
9
+ widget:
10
+ - text: "Files encrypted with ransom demand for Bitcoin payment"
11
+ example_title: "Ransomware Example"
12
+ - text: "Website unreachable due to massive spike in incoming requests"
13
+ example_title: "DDoS Example"
14
+ - text: "Employee downloading sensitive customer data before resignation"
15
+ example_title: "Insider Threat Example"
16
+ ---
17
+
18
+ # Cybersecurity BERT Classifier
19
+
20
+ This model is a fine-tuned `bert-base-uncased` model that classifies cybersecurity alerts into five threat categories.
21
+
22
+ ## Model Details
23
+
24
+ - **Base Model:** `bert-base-uncased`
25
+ - **Task:** Text Classification
26
+ - **Labels:**
27
+ - `0`: Ransomware
28
+ - `1`: DDoS
29
+ - `2`: Insider Threat
30
+ - `3`: Web Attack
31
+ - `4`: Benign
32
+
33
+ ## Intended Uses & Limitations
34
+
35
+ This model is intended for security operations center (SOC) teams to automatically triage and classify security alert text. It achieves **92.86% accuracy** on a curated test set.
36
+
37
+ ## How to Use
38
+
39
+ You can use this model directly with the Transformers pipeline for text classification:
40
+
41
+ ```python
42
+ from transformers import pipeline
43
+
44
+ classifier = pipeline("text-classification", model="Aikaksh-Singh-Routela/cybersecurity-bert-model")
45
+
46
+ result = classifier("Files encrypted with ransom demand for Bitcoin payment")
47
+ print(result)
48
+ # Expected output: [{'label': 'Ransomware', 'score': 0.9286}]
49
+
50
+
51
+
52
+ ---
53
+
54
+ ### ✅ Why This Matters
55
+
56
+ | Without a Model Card | With a Model Card |
57
+ | :--- | :--- |
58
+ | Looks incomplete or abandoned. | Looks professional and well-documented. |
59
+ | Hard for others (or recruiters) to know what it does. | Anyone can instantly see its purpose, performance, and how to use it. |
60
+ | Poor discoverability. | Can be found through searches for relevant tags like "cybersecurity" or "text-classification." |
61
+
62
+ After adding this, your model page will be complete and will effectively showcase your project. You have done excellent work; this is the final polish to make it shine.